train.json example update

train.json

@@ -1,36 +1,36 @@
 {
-  "resume_ckpt": "sd_v1-5_vae",
-  "lr_scheduler": "cosine",
-  "lr_warmup_steps": null,
-  "lr_decay_steps": 0,
-  "log_step": 25,
-  "max_epochs": 50,
-  "ckpt_every_n_minutes": null,
-  "save_every_n_epochs": 20,
-  "lr": 4.5e-06,
-  "useadam8bit": true,
-  "project_name": "my_project",
-  "sample_prompts": "sample_prompts.txt",
-  "sample_steps": 300,
-  "disable_textenc_training": false,
-  "batch_size": 7,
-  "clip_grad_norm": null,
-  "grad_accum": 5,
-  "clip_skip": 0,
-  "data_root": "X:\\mytrainingdata\\my_project_stuff",
-  "wandb": false,
-  "save_optimizer": false,
-  "resolution": 512,
   "amp": false,
+  "batch_size": 10,
+  "ckpt_every_n_minutes": null,
+  "clip_grad_norm": null,
+  "clip_skip": 0,
   "cond_dropout": 0.04,
-  "logdir": "logs",
-  "save_ckpt_dir": null,
-  "scale_lr": false,
-  "seed": 555,
+  "data_root": "X:\\my_project_data\\project_abc",
+  "disable_textenc_training": false,
+  "disable_xformers": false,
   "flip_p": 0.0,
   "gpuid": 0,
-  "write_schedule": true,
   "gradient_checkpointing": true,
-  "disable_xformers": false,
-  "lowvram": false
+  "grad_accum": 1,
+  "logdir": "logs",
+  "log_step": 25,
+  "lowvram": false,
+  "lr": 3.5e-06,
+  "lr_decay_steps": 0,
+  "lr_scheduler": "constant",
+  "lr_warmup_steps": null,
+  "max_epochs": 30,
+  "project_name": "project_abc",
+  "resolution": 512,
+  "resume_ckpt": "sd_v1-5_vae",
+  "sample_prompts": "sample_prompts.txt",
+  "sample_steps": 300,
+  "save_ckpt_dir": null,
+  "save_every_n_epochs": 20,
+  "save_optimizer": false,
+  "scale_lr": false,
+  "seed": 555,
+  "useadam8bit": true,
+  "wandb": false,
+  "write_schedule": false
 }

train.py

@@ -110,7 +110,7 @@ def setup_local_logger(args):
     json_config = json.dumps(vars(args), indent=2)
     datetimestamp = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
 
-    with open(os.path.join(log_path, f"{args.project_name}-{datetimestamp}.json"), "w") as f:
+    with open(os.path.join(log_path, f"{args.project_name}-{datetimestamp}_cfg.json"), "w") as f:
         f.write(f"{json_config}")
 
     logfilename = os.path.join(log_path, f"{args.project_name}-{datetimestamp}.log")