diffumatch_model / training_options.json
daidedou's picture
Add Json file
11b952d verified
{
"misc": {
"cuda": true,
"device": 0,
"checkpoint_interval": 1,
"log_interval": 812,
"desc": null,
"precond": false,
"dry_run": false
},
"data": {
"root_dir": "dfaust_fmap_30x30_clean.npy",
"name": "DFAUST_fmap_30",
"n_fmap": 30,
"out": "fmap_exps",
"cond": false,
"in_memory": "RAM",
"normalize": true,
"abs": true,
"run": "fmap_exps/00053-DFAUST_fmap_30abs-uncond-DiT-gpus1-batch256-fp32"
},
"add_name": {
"do": false,
"name": "bis"
},
"architecture": {
"model": "DiT",
"name_arch": "DiT-S/4",
"input_type": "img",
"cond": false
},
"hyper_params": {
"iterations": 200,
"batch_size": 256,
"lr": 0.001,
"lr_rampup_kimg": 10000,
"ema_halflife_nshape": 500,
"ema_rampup_ratio": 0.05,
"dropout": 0,
"loss_name": "VPLoss",
"ls": 1
},
"perfs": {
"fp16": false,
"workers": 1
},
"resume": {
"pkl": null,
"transfer": null,
"kimg_per_tick": 5,
"snapshot_ticks": 50,
"state_dump_ticks": 50
},
"datadir": "./data",
"device": "cuda:0"
}