GaumlessGraham
/

7inchInner1730NoShift

Model card Files Files and versions

Metrics Training metrics Community

GaumlessGraham commited on Sep 15, 2024

Commit

0dace64

·

verified ·

1 Parent(s): 922d0ee

Update inner.py

Files changed (1) hide show

inner.py +2 -2

inner.py CHANGED Viewed

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

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