From 1e207677f3cfa9ec4d24e75e9b582d004d83264c Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Thu, 26 Jan 2023 12:07:28 -0500 Subject: [PATCH] sd21 example json config --- trainSD21.json | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 trainSD21.json diff --git a/trainSD21.json b/trainSD21.json new file mode 100644 index 0000000..66ce834 --- /dev/null +++ b/trainSD21.json @@ -0,0 +1,40 @@ +{ + "amp": true, + "batch_size": 6, + "ckpt_every_n_minutes": null, + "clip_grad_norm": null, + "clip_skip": 2, + "cond_dropout": 0.04, + "data_root": "X:\\my_project_data\\project_abc", + "disable_textenc_training": false, + "disable_xformers": false, + "flip_p": 0.0, + "gpuid": 0, + "gradient_checkpointing": true, + "grad_accum": 1, + "logdir": "logs", + "log_step": 25, + "lowvram": false, + "lr": 0.6e-06, + "lr_decay_steps": 0, + "lr_scheduler": "constant", + "lr_warmup_steps": null, + "max_epochs": 30, + "notebook": false, + "project_name": "project_abc_sd21", + "resolution": 768, + "resume_ckpt": "v2-1_768-nonema-pruned", + "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, + "shuffle_tags": false, + "useadam8bit": true, + "wandb": false, + "write_schedule": false, + "rated_dataset": false, + "rated_dataset_target_dropout_rate": 50 +}