train-diffuser

Paused

nroggendorff commited on Jul 20, 2024

Commit

c90d126

verified ·

1 Parent(s): d1a06b7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,7 +102,6 @@ args = TrainingArguments(
     per_device_train_batch_size=64,
     gradient_accumulation_steps=4,
     learning_rate=1e-5,
-    save_steps=100000,
     fp16=True,
     optim="sgd",
     optim_target_modules=["attn", "mlp"]

     per_device_train_batch_size=64,
     gradient_accumulation_steps=4,
     learning_rate=1e-5,
     fp16=True,
     optim="sgd",
     optim_target_modules=["attn", "mlp"]