Merge pull request #148 from damian0815/fix_default_sd21_clipskip
fix clip_skip default in trainSD21.json and update other params
This commit is contained in:
commit
79886981a2
|
@ -3,7 +3,7 @@
|
||||||
"batch_size": 6,
|
"batch_size": 6,
|
||||||
"ckpt_every_n_minutes": null,
|
"ckpt_every_n_minutes": null,
|
||||||
"clip_grad_norm": null,
|
"clip_grad_norm": null,
|
||||||
"clip_skip": 2,
|
"clip_skip": 0,
|
||||||
"cond_dropout": 0.04,
|
"cond_dropout": 0.04,
|
||||||
"data_root": "X:\\my_project_data\\project_abc",
|
"data_root": "X:\\my_project_data\\project_abc",
|
||||||
"disable_textenc_training": false,
|
"disable_textenc_training": false,
|
||||||
|
@ -21,20 +21,23 @@
|
||||||
"lr_warmup_steps": null,
|
"lr_warmup_steps": null,
|
||||||
"max_epochs": 30,
|
"max_epochs": 30,
|
||||||
"notebook": false,
|
"notebook": false,
|
||||||
|
"optimizer_config": "optimizer.json",
|
||||||
"project_name": "project_abc_sd21",
|
"project_name": "project_abc_sd21",
|
||||||
"resolution": 768,
|
"resolution": 768,
|
||||||
"resume_ckpt": "v2-1_768-nonema-pruned",
|
"resume_ckpt": "v2-1_768-nonema-pruned",
|
||||||
"sample_prompts": "sample_prompts.txt",
|
"sample_prompts": "sample_prompts.txt",
|
||||||
"sample_steps": 300,
|
"sample_steps": 300,
|
||||||
"save_ckpt_dir": null,
|
"save_ckpt_dir": null,
|
||||||
|
"save_ckpts_from_n_epochs": 0,
|
||||||
"save_every_n_epochs": 20,
|
"save_every_n_epochs": 20,
|
||||||
"save_optimizer": false,
|
"save_optimizer": false,
|
||||||
"scale_lr": false,
|
"scale_lr": false,
|
||||||
"seed": 555,
|
"seed": 555,
|
||||||
"shuffle_tags": false,
|
"shuffle_tags": false,
|
||||||
"useadam8bit": true,
|
"validation_config": "validation_default.json",
|
||||||
"wandb": false,
|
"wandb": false,
|
||||||
"write_schedule": false,
|
"write_schedule": false,
|
||||||
"rated_dataset": false,
|
"rated_dataset": false,
|
||||||
"rated_dataset_target_dropout_rate": 50
|
"rated_dataset_target_dropout_percent": 50,
|
||||||
|
"zero_frequency_noise_ratio": 0.02
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue