GOVINDFROM's picture
Recurrent masked diffusion: results, code, card
ae02605 verified
Raw History Blame Contribute Delete
737 Bytes
{
"args": {
"mode": "mdlm",
"objective": "ar",
"K": 8,
"lam": 1.0,
"pe": "ape",
"recurrent": false,
"R": 32,
"no_deep_sup": false,
"R_infer": 0,
"root": "data/sudoku",
"hard": "/scratch/zt1/project/msml612/user/govind02/pcmd/data/sudoku_hard",
"augment": true,
"d": 128,
"layers": 12,
"heads": 4,
"bs": 256,
"steps": 120000,
"lr": 0.0003,
"warmup": 3000,
"seed": 0,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/base-ar-s0"
},
"decode": {
"ar": {
"board": 0.0,
"cell": 0.16823683003422604,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 81.0
}
}
}