code/configs/dmad.json

{
 "method": "dmad",
 "out_dir": "/workspace/runs/dmad",
 "teacher_dir": ["/workspace/teacher", "/workspace/teacher2"],
 "iterations": 2500,
 "micro": 2,
 "critic_warmup": 50,
 "lr_gen": 2e-5, "lr_time": 2e-5, "lr_q": 2e-5,
 "lr_critic": 4e-5,
 "ema": 0.998,
 "log_every": 10,
 "sample_every": 100,
 "save_every": 200,
 "export_every": 500,
 "eval_count": 8
}