config.json

{
  "train": {
    "data_repo": "wuff-mann/ATLAS-WDS",
    "n_freqs": 47,
    "n_dirs": 72,
    "freq_min": 0.02,
    "freq_max": 0.5,
    "hs_ceiling": 25.0,
    "extreme_hs": 8.0,
    "extreme_npeaks": 4,
    "quick_verify": true,
    "target_samples": 40000,
    "normal_ratio": 0.65,
    "val_ratio": 0.1,
    "epochs": 50,
    "batch": 64,
    "lr": 0.0003,
    "weight_decay": 0.0001,
    "grad_clip": 5.0,
    "beta_warmup": 25,
    "num_workers": 4,
    "out_dir": "./cwa_out_v2",
    "model_repo": "wuff-mann/CNN-WaveAttention-v2",
    "ckpt_every": 20,
    "klsnap_every": 5,
    "upload_every": 10,
    "seed": 42
  },
  "model": {
    "n_freqs": 47,
    "n_dirs": 72,
    "pad_freqs": 48,
    "enc_channels": [
      32,
      64,
      128
    ],
    "feat_grid": [
      12,
      18
    ],
    "dec_grid": [
      6,
      9
    ],
    "n_slots": 6,
    "slot_dim": 64,
    "latent_dim": 16,
    "slot_iters": 3,
    "dec_channels": [
      128,
      64,
      32
    ],
    "energy_compose": true,
    "recon_mse_w": 0.5,
    "recon_l1_w": 0.5,
    "beta_max": 0.0003,
    "free_bits": 0.0,
    "mask_entropy_w": 0.0,
    "anti_dominate_w": 0.01,
    "usage_cap": 0.6,
    "diversity_w": 0.01,
    "diversity_on": "mask"
  }
}