{
  "config": {
    "run": {
      "name": "us_wildfire_size",
      "seed": 0,
      "dtype": "float32"
    },
    "dataset": {
      "kind": "tabular",
      "name": "us_wildfire_size",
      "file": "us_wildfire_size.pt",
      "split": {
        "standardize": true
      }
    },
    "network": {
      "name": "mlp",
      "params": {
        "width": 128,
        "depth": 3,
        "time_dim": 64,
        "dropout": 0.0,
        "use_norm": true
      }
    },
    "models": {
      "ddpm_v": {
        "params": {
          "n_steps": 128,
          "sigma_max": 2.0,
          "sampler": "ddpm"
        }
      },
      "dlpm_eps": {
        "params": {
          "n_steps": 128,
          "alpha": 1.8,
          "sampler": "native"
        }
      },
      "tedm_origin": {
        "params": {
          "n_steps": 64,
          "nu": 2.2,
          "sigma_min": 0.005,
          "sigma_max": 5.0,
          "sigma_data": 1.0,
          "solver": "edm_stochastic_heun"
        }
      }
    },
    "pilot": {
      "n_trials": 2,
      "train": {
        "n_epochs": 96,
        "early_stopping_patience": 32,
        "freq_logging": 16,
        "stats_freq_epochs": 1
      },
      "sweep": {
        "ddpm_v": {
          "model": {
            "sigma_max": [
              2.0,
              5.0
            ]
          },
          "train": {
            "lr": [
              0.0002,
              0.0005
            ]
          }
        },
        "dlpm_eps": {
          "train": {
            "lr": [
              0.0002,
              0.0005
            ]
          }
        },
        "tedm_origin": {
          "train": {
            "lr": [
              0.0002,
              0.0005
            ]
          }
        }
      }
    },
    "train": {
      "device": "cuda",
      "data_device": "cpu",
      "batch_size": 512,
      "n_epochs": 640,
      "early_stopping_patience": 100,
      "lr": 0.0005,
      "num_workers": 0,
      "use_adamw": true,
      "weight_decay": 1e-06,
      "grad_clip_norm": 10.0,
      "lr_schedule": "cosine",
      "warmup_steps": 100,
      "cosine_eta_min_ratio": 0.05,
      "freq_logging": 16,
      "stats_freq_epochs": 1,
      "log_grad_norm": true,
      "ckpt_freq_epochs": 64,
      "ckpt_keep_last": 4
    },
    "test": {
      "batch_size": 512,
      "loss_repeats": 2,
      "loss_samples": 2000,
      "n_samples": 32
    },
    "huggingface": {
      "namespace": "Diffusion-Research-Lab",
      "repository": "us-wildfire-size-1d-diffusion"
    }
  },
  "model_name": "ddim_v",
  "model_params": {
    "n_steps": 128,
    "sigma_max": 2.0,
    "sampler": "ddim",
    "eta": 0.0,
    "sample_steps": null
  },
  "sample_shape": [
    1
  ],
  "data_scale": 0.03256954997777939
}