GOVINDFROM's picture
Recurrent masked diffusion: results, code, card
ae02605 verified
Raw History Blame Contribute Delete
3.11 kB
{
"args": {
"mode": "mdlm",
"objective": "direct",
"K": 8,
"lam": 1.0,
"pe": "ape",
"recurrent": true,
"R": 32,
"no_deep_sup": false,
"R_infer": 64,
"root": "data/sudoku",
"hard": "/scratch/zt1/project/msml612/user/govind02/pcmd/data/sudoku_hard",
"augment": true,
"d": 128,
"layers": 1,
"heads": 4,
"bs": 256,
"steps": 120000,
"lr": 0.0003,
"warmup": 3000,
"seed": 1,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/base-directrec-s1"
},
"decode": {
"fixedK_1": {
"board": 0.015,
"cell": 0.7376415793373557,
"valid_sudoku": 0.015,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.009,
"cell": 0.7314314950072334,
"valid_sudoku": 0.009,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.003,
"cell": 0.7721146042835468,
"valid_sudoku": 0.003,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.005,
"cell": 0.7732437105253872,
"valid_sudoku": 0.005,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.0,
"cell": 0.7650753325570728,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.002,
"cell": 0.7653223245474754,
"valid_sudoku": 0.002,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.003,
"cell": 0.7670689107653224,
"valid_sudoku": 0.003,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.008,
"cell": 0.7843759923785328,
"valid_sudoku": 0.008,
"clues_kept": 1.0,
"nfe": 53.75
},
"entbudget_0.05": {
"board": 0.037,
"cell": 0.7805652588123214,
"valid_sudoku": 0.037,
"clues_kept": 1.0,
"nfe": 46.0
},
"entbudget_0.2": {
"board": 0.06,
"cell": 0.7796831445608835,
"valid_sudoku": 0.06,
"clues_kept": 1.0,
"nfe": 39.0
},
"entbudget_0.5": {
"board": 0.063,
"cell": 0.7810592427931266,
"valid_sudoku": 0.063,
"clues_kept": 1.0,
"nfe": 29.5
},
"entbudget_1.0": {
"board": 0.064,
"cell": 0.7823647718852546,
"valid_sudoku": 0.064,
"clues_kept": 1.0,
"nfe": 22.75
},
"entbudget_2.0": {
"board": 0.057,
"cell": 0.7639462263152323,
"valid_sudoku": 0.057,
"clues_kept": 1.0,
"nfe": 15.0
},
"conf_0.9": {
"board": 0.061,
"cell": 0.7680039518718464,
"valid_sudoku": 0.061,
"clues_kept": 1.0,
"nfe": 24.5
},
"conf_0.99": {
"board": 0.066,
"cell": 0.7839702198228714,
"valid_sudoku": 0.066,
"clues_kept": 1.0,
"nfe": 40.0
},
"conf_0.999": {
"board": 0.057,
"cell": 0.7903920115733389,
"valid_sudoku": 0.057,
"clues_kept": 1.0,
"nfe": 49.25
}
},
"best_board": 0.066
}