config.json

{
  "data_root": "/home/ubuntu/work/BCI_dataset",
  "source_domain": "HE",
  "target_domain": "IHC",
  "image_size": 256,
  "train_augment": true,
  "base_channels": 192,
  "channel_mults": "1,1,2,2,4,4",
  "attn_resolutions": "32,16,8",
  "num_heads": 4,
  "num_train_timesteps": 1000,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "he_gen_prob": 0.5,
  "batch_size": 24,
  "max_iters": 3000,
  "lr": 0.0001,
  "ema_decay": 0.999,
  "grad_clip": 1.0,
  "amp": "bf16",
  "seed": 0,
  "save_every": 1000,
  "output_dir": "/home/ubuntu/work/dualtask_rot",
  "resume": null,
  "num_workers": 4,
  "diag_scalar_every": 20,
  "diag_layer_every": 500,
  "diag_sample_every": 500,
  "diag_per_t_every": 500,
  "diag_disable_activations": false
}