Spaces:

JoPmt
/

Bedtime_Short_Stories

Runtime error

JoPmt commited on Mar 14, 2024

Commit

fab4ad3

verified ·

1 Parent(s): d8e2bf7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,9 +20,9 @@ data_collator = DataCollatorForLanguageModeling(
 training_args = TrainingArguments(
     output_dir=cwd,
     overwrite_output_dir=True,
-    num_train_epochs=6,
     per_device_train_batch_size=8,
-    save_steps=500,
     save_total_limit=1,
 )
 trainer = Trainer(

 training_args = TrainingArguments(
     output_dir=cwd,
     overwrite_output_dir=True,
+    num_train_epochs=5,
     per_device_train_batch_size=8,
+    save_steps=1000,
     save_total_limit=1,
 )
 trainer = Trainer(