| { | |
| "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 | |
| } |