Spaces:

DelaliScratchwerk
/

time-period-api

Sleeping

DelaliScratchwerk commited on 21 days ago

Commit

952742f

verified ·

1 Parent(s): c160ba6

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -115,8 +115,8 @@ training_args = TrainingArguments(
     output_dir="out",
     per_device_train_batch_size=16,
     per_device_eval_batch_size=32,
-    learning_rate=2e-5,
-    num_train_epochs=4,
     eval_strategy="epoch",
     save_strategy="no",
     load_best_model_at_end=False,

     output_dir="out",
     per_device_train_batch_size=16,
     per_device_eval_batch_size=32,
+    learning_rate=5e-5,
+    num_train_epochs=10,
     eval_strategy="epoch",
     save_strategy="no",
     load_best_model_at_end=False,