cosmopolitan / 3 - model fine-tuning /2 - vodka_v5_768_adamw8bit /vodka_v5_4_768-20231011-114741_cfg.json
pxovela's picture
fine-tuning
196ee88
{
"config": "train.json",
"batch_size": 4,
"ckpt_every_n_minutes": null,
"clip_grad_norm": null,
"clip_skip": 0,
"cond_dropout": 0.05,
"data_root": "mj_raw/training_data",
"disable_amp": false,
"disable_textenc_training": false,
"disable_xformers": false,
"flip_p": 0.0,
"gpuid": 0,
"gradient_checkpointing": true,
"grad_accum": 1,
"logdir": "logs",
"log_step": 1,
"lowvram": false,
"lr_decay_steps": 0,
"lr_scheduler": "constant",
"lr_warmup_steps": null,
"max_epochs": 100,
"notebook": false,
"optimizer_config": "optimizer_dadapt.json",
"project_name": "vodka_v5_4_768",
"resolution": 768,
"resume_ckpt": "sd_v1-5_vae",
"run_name": null,
"sample_prompts": "sample_prompts.txt",
"sample_steps": 10000000,
"save_ckpt_dir": null,
"save_ckpts_from_n_epochs": 0,
"save_every_n_epochs": 10,
"save_optimizer": false,
"scale_lr": false,
"seed": 555,
"shuffle_tags": false,
"validation_config": "validation_default.json",
"wandb": true,
"write_schedule": false,
"rated_dataset": false,
"rated_dataset_target_dropout_percent": 50,
"zero_frequency_noise_ratio": 0.01,
"enable_zero_terminal_snr": false,
"load_settings_every_epoch": false,
"min_snr_gamma": null,
"ema_decay_rate": null,
"ema_strength_target": null,
"ema_update_interval": null,
"ema_device": null,
"ema_sample_nonema_model": false,
"ema_sample_ema_model": false,
"ema_resume_model": null,
"amp": true,
"disable_unet_training": false,
"lr": null,
"no_prepend_last": false,
"no_save_ckpt": false,
"plugins": null,
"save_full_precision": false,
"keep_tags": 0,
"useadam8bit": false
}