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