2023-01-03 12:27:26 -07:00
|
|
|
{
|
2023-01-17 17:42:07 -07:00
|
|
|
"amp": true,
|
2023-01-03 13:10:57 -07:00
|
|
|
"batch_size": 10,
|
2023-01-03 12:27:26 -07:00
|
|
|
"ckpt_every_n_minutes": null,
|
|
|
|
|
"clip_grad_norm": null,
|
|
|
|
|
"clip_skip": 0,
|
|
|
|
|
"cond_dropout": 0.04,
|
2023-01-03 13:10:57 -07:00
|
|
|
"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,
|
2023-01-03 12:27:26 -07:00
|
|
|
"logdir": "logs",
|
2023-01-03 13:10:57 -07:00
|
|
|
"log_step": 25,
|
|
|
|
|
"lowvram": false,
|
2023-01-17 17:42:07 -07:00
|
|
|
"lr": 1.5e-06,
|
2023-01-03 13:10:57 -07:00
|
|
|
"lr_decay_steps": 0,
|
|
|
|
|
"lr_scheduler": "constant",
|
|
|
|
|
"lr_warmup_steps": null,
|
|
|
|
|
"max_epochs": 30,
|
2023-01-15 20:07:37 -07:00
|
|
|
"notebook": false,
|
2023-01-03 13:10:57 -07:00
|
|
|
"project_name": "project_abc",
|
|
|
|
|
"resolution": 512,
|
|
|
|
|
"resume_ckpt": "sd_v1-5_vae",
|
|
|
|
|
"sample_prompts": "sample_prompts.txt",
|
|
|
|
|
"sample_steps": 300,
|
2023-01-03 12:27:26 -07:00
|
|
|
"save_ckpt_dir": null,
|
2023-01-03 13:10:57 -07:00
|
|
|
"save_every_n_epochs": 20,
|
|
|
|
|
"save_optimizer": false,
|
2023-01-03 12:27:26 -07:00
|
|
|
"scale_lr": false,
|
|
|
|
|
"seed": 555,
|
2023-01-06 17:12:52 -07:00
|
|
|
"shuffle_tags": false,
|
2023-01-03 13:10:57 -07:00
|
|
|
"useadam8bit": true,
|
|
|
|
|
"wandb": false,
|
2023-01-14 06:00:30 -07:00
|
|
|
"write_schedule": false,
|
|
|
|
|
"rated_dataset": false,
|
2023-01-31 17:21:49 -07:00
|
|
|
"rated_dataset_target_dropout_percent": 50
|
2023-01-14 06:00:30 -07:00
|
|
|
}
|
