config.json

{
    "training": {
        "batch_size": 8,
        "n_epochs": 200001,
        "likelihood_weighting": false,
        "reduce_mean": false,
        "sde": "vesde",
        "continuous": true
    },
    "sampling": {
        "snr": 0.075,
        "correct_steps": 0,
        "batch_size": 1,
        "num_samples": 25
    },
    "data": {
        "path": "/mnt/home/rlegin/ceph/datasets/",
        "image_size": 128,
        "num_input_channels": 2,
        "num_output_channels": 1,
        "noise_sigma": 0.1,
        "cosmo_ids": [20, 40, 50, 70, 80, 90]
    },
    "model": {
        "sigma_max": 100,
        "sigma_min": 0.01,
        "num_scales": 1000,
        "dropout": 0.1,
        "embedding_type": "fourier",
        "sampling_eps": 1e-5,
        "T": 1.0,
        "workdir": "run/",
        "ema_rate": 0.999,
        "normalization": "GroupNorm",
        "nonlinearity": "swish",
        "nf": 32,
        "ch_mult": [1,2, 2, 1, 1],
        "num_res_blocks": 2,
        "resamp_with_conv": true,
        "conditional": true,
        "fir": false,
        "fir_kernel": [1, 3, 3, 1],
        "skip_rescale": true,
        "init_scale": 0.0,
        "fourier_scale": 16,
        "conv_size": 3
    },
    "optim": {
        "weight_decay": 0,
        "optimizer": "Adam",
        "lr": 2e-4,
        "beta1": 0.9,
        "eps": 1e-8,
        "warmup": 5000,
        "grad_clip": 1.0
    },
    "seed": 42,
    "device": "cuda:0"
}