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