config new code sweep

---------------------------------------
Begin Slurm Prolog: Jan-06-2026 13:09:31
Job ID:    3166794
User ID:   echen324
Account:   gts-yxie77-paid
Job name:  lfm_cifar_sweep
Partition: gpu-h200
QOS:       inferno
---------------------------------------
base_dir: results_cifar
data:
  channels: 3
  dataname: cifar10
  image_shape:
  - 32
  - 32
  num_workers: 2
  pin_memory: true
  root: ./data
eval:
  cal_fid: true
  fid_num_samples: 50000
  gen_bs: 512
model:
  L: 3
  alpha: 1.0
  beta: 1.0
  beta_sample_t: true
  coupling: dependent
  dropout: 0
  hidden_dim: 128
  hks:
  - 0.1
  - 0.15
  - 0.225
  - null
  interp: linear
  num_head_channels: 32
  res_depth: 2
  unet_dim_mults:
  - 1
  - 2
  - 2
  - 2
ode:
  method: dopri5
  tol: 1e-3
save_dir: cifar_N4
training:
  batch_push_size: 512
  batch_size: 256
  ema_decay: 0.999
  eval_only: false
  flip_aug: true
  load_ema: true
  lr: 0.0001
  max_batch: 50000
  resume: false
  use_grad_clip: true
  warm_start: true
  warmup: 5000
visualize:
  viz_freq: 10000

[INFO] Random seed set to 0
Saving results to: results_cifar/cifar_N4_seed0
Unet: base channels: 128, dim_mults: [1, 2, 2, 2],             res depth: 2, head channels: 32
beta sample enabled: 1.0, 1.0
ODE config:
{'method': 'dopri5', 'tol': '1e-3'}
####### Block 1 with hk = 0.1
Number of parameters: 35746.307k