Spaces:

JoPmt
/

Bedtime_Short_Stories

Runtime error

JoPmt commited on Jan 3, 2024

Commit

ab8ec52

1 Parent(s): 7776078

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ data_collator = DataCollatorForLanguageModeling(
 training_args = TrainingArguments(
     output_dir=cwd,
     overwrite_output_dir=True,
-    num_train_epochs=5,
     per_device_train_batch_size=8,
     save_steps=500,
     save_total_limit=1,

 training_args = TrainingArguments(
     output_dir=cwd,
     overwrite_output_dir=True,
+    num_train_epochs=7,
     per_device_train_batch_size=8,
     save_steps=500,
     save_total_limit=1,