ketanchaudhary88 commited on
Commit
2d7b37e
Β·
verified Β·
1 Parent(s): a9dfd01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -53,14 +53,14 @@ max_steps = steps_per_epoch * num_train_epochs
53
 
54
  # Define training arguments
55
  training_args = TrainingArguments(
56
- output_dir='./results',
57
- evaluation_strategy='epoch', # Evaluate at the end of each epoch
58
  learning_rate=2e-5,
59
  per_device_train_batch_size=8, # You can adjust batch size here
60
  per_device_eval_batch_size=8,
61
  num_train_epochs=num_train_epochs, # Setting epochs to 3
62
  weight_decay=0.01,
63
- logging_dir='./logs',
64
  logging_steps=500,
65
  save_steps=1000,
66
  load_best_model_at_end=True,
 
53
 
54
  # Define training arguments
55
  training_args = TrainingArguments(
56
+ output_dir='results',
57
+ eval_strategy='epoch', # Evaluate at the end of each epoch
58
  learning_rate=2e-5,
59
  per_device_train_batch_size=8, # You can adjust batch size here
60
  per_device_eval_batch_size=8,
61
  num_train_epochs=num_train_epochs, # Setting epochs to 3
62
  weight_decay=0.01,
63
+ logging_dir='logs',
64
  logging_steps=500,
65
  save_steps=1000,
66
  load_best_model_at_end=True,