rico_diffusion-20250309-203322_main.json

{
  "config": "rico_diffusion.json",
  "attn_type": "sdp",
  "batch_size": 7,
  "ckpt_every_n_minutes": 1000000000.0,
  "clip_grad_norm": null,
  "clip_skip": 2,
  "cond_dropout": 0.1,
  "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,
  "amp": true,
  "disable_unet_training": false,
  "embedding_perturbation": 0.0,
  "lr": null,
  "lr_decay_steps": 124140,
  "lr_scheduler": "constant",
  "lr_warmup_steps": 2482,
  "no_prepend_last": false,
  "no_save_ckpt": false,
  "plugins": null,
  "run_name": null,
  "save_full_precision": false,
  "timestep_start": 0,
  "timestep_end": 1000,
  "train_sampler": "ddpm",
  "wandb": false,
  "write_schedule": false,
  "rated_dataset": false,
  "rated_dataset_target_dropout_percent": 50,
  "load_settings_every_epoch": null,
  "ema_decay_rate": null,
  "ema_strength_target": null,
  "ema_update_interval": 500,
  "ema_device": "cpu",
  "ema_sample_nonema_model": false,
  "ema_sample_ema_model": false,
  "ema_resume_model": null,
  "pyramid_noise_discount": null,
  "aspects": [
    [
      512,
      512
    ],
    [
      576,
      448
    ],
    [
      448,
      576
    ],
    [
      640,
      384
    ],
    [
      384,
      640
    ],
    [
      768,
      320
    ],
    [
      320,
      768
    ],
    [
      896,
      256
    ],
    [
      256,
      896
    ],
    [
      1024,
      256
    ],
    [
      256,
      1024
    ]
  ]
}