train-diffuser

Paused

nroggendorff commited on Jul 17, 2024

Commit

e1e5f7f

verified ·

1 Parent(s): c987618

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -99,7 +99,7 @@ args = TrainingArguments(
     output_dir="mayo",
     num_train_epochs=1,
     gradient_accumulation_steps=4,
-    per_device_train_batch_size=1,
     learning_rate=1e-5,
     save_steps=100000,
     fp16=True,

     output_dir="mayo",
     num_train_epochs=1,
     gradient_accumulation_steps=4,
+    per_device_train_batch_size=8,
     learning_rate=1e-5,
     save_steps=100000,
     fp16=True,