GaumlessGraham commited on
Commit
b1790a3
·
verified ·
1 Parent(s): cef8f00

Update inner.py

Browse files
Files changed (1) hide show
  1. inner.py +2 -2
inner.py CHANGED
@@ -63,8 +63,8 @@ class TrainingConfig:
63
  gradient_accumulation_steps = 1
64
  learning_rate = 1e-4
65
  lr_warmup_steps = 250
66
- save_image_epochs = 600
67
- save_model_epochs = 600
68
  mixed_precision = "fp16" # `no` for float32, `fp16` for automatic mixed precision
69
  output_dir = "Inner1730_10Real" # the model name locally and on the HF Hub
70
 
 
63
  gradient_accumulation_steps = 1
64
  learning_rate = 1e-4
65
  lr_warmup_steps = 250
66
+ save_image_epochs = 500
67
+ save_model_epochs = 500
68
  mixed_precision = "fp16" # `no` for float32, `fp16` for automatic mixed precision
69
  output_dir = "Inner1730_10Real" # the model name locally and on the HF Hub
70