Spaces:

JoPmt
/

Bedtime_Short_Stories

Runtime error

JoPmt commited on Mar 14, 2024

Commit

d8e2bf7

verified ·

1 Parent(s): ab8ec52

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=7,
     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=6,
     per_device_train_batch_size=8,
     save_steps=500,
     save_total_limit=1,