GaumlessGraham
/

Outer1730_10Real

Model card Files Files and versions

Metrics Training metrics Community

GaumlessGraham commited on Sep 19, 2024

Commit

2f98e0a

·

verified ·

1 Parent(s): 8924c24

Update outer.py

Files changed (1) hide show

outer.py +2 -2

outer.py CHANGED Viewed

@@ -63,8 +63,8 @@ class TrainingConfig:
     gradient_accumulation_steps = 1
     learning_rate = 1e-4
     lr_warmup_steps = 250
-    save_image_epochs = 450
-    save_model_epochs = 450
     mixed_precision = "fp16"  # `no` for float32, `fp16` for automatic mixed precision
     output_dir = "Outer1730_10Real"  # the model name locally and on the HF Hub

     gradient_accumulation_steps = 1
     learning_rate = 1e-4
     lr_warmup_steps = 250
+    save_image_epochs = 550
+    save_model_epochs = 550
     mixed_precision = "fp16"  # `no` for float32, `fp16` for automatic mixed precision
     output_dir = "Outer1730_10Real"  # the model name locally and on the HF Hub