File size: 1,097 Bytes
d045179 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | {
"attn_type": "sdp",
"batch_size": 7,
"ckpt_every_n_minutes": null,
"clip_grad_norm": null,
"clip_skip": 2,
"cond_dropout": 0.10,
"data_root": "/home/azhure/Tesis/save_data",
"disable_amp": false,
"disable_textenc_training": false,
"flip_p": 0.0,
"gpuid": 0,
"gradient_checkpointing": true,
"grad_accum": 1,
"logdir": "logs",
"log_step": 25,
"loss_type": "huber",
"max_epochs": 5,
"optimizer_config": "optimizer_rico.json",
"project_name": "rico_diffusion",
"resolution": 512,
"resume_ckpt": "logs/rico_diffusion_v2_comp_big-20250307-090306/ckpts/last-rico_diffusion_v2_comp_big-ep04-gs32932",
"sample_prompts": "sample_prompts_rico.json",
"sample_steps": 15000,
"save_ckpt_dir": "model_output",
"save_ckpts_from_n_epochs": 0,
"save_every_n_epochs": 5,
"save_optimizer": false,
"scale_lr": false,
"seed": 555,
"keep_tags": 3,
"shuffle_tags": false,
"validation_config": "validation_rico_off.json",
"zero_frequency_noise_ratio": 0.02,
"min_snr_gamma": 5.0,
"enable_zero_terminal_snr": true
} |