GOVINDFROM's picture
Recurrent masked diffusion: results, code, card
ae02605 verified
Raw History Blame Contribute Delete
3.08 kB
{
"args": {
"mode": "full",
"K": 8,
"lam": 1.0,
"pe": "ape",
"recurrent": true,
"R": 32,
"no_deep_sup": true,
"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": 0,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/abl-nodeepsup-s0"
},
"decode": {
"fixedK_1": {
"board": 0.932,
"cell": 0.9677146183973748,
"valid_sudoku": 0.932,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.935,
"cell": 0.9694964891852793,
"valid_sudoku": 0.935,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.94,
"cell": 0.9735012878868071,
"valid_sudoku": 0.94,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.952,
"cell": 0.9796408030768146,
"valid_sudoku": 0.952,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.957,
"cell": 0.9813344624395752,
"valid_sudoku": 0.957,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.964,
"cell": 0.9835926749232561,
"valid_sudoku": 0.964,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.97,
"cell": 0.985268692000988,
"valid_sudoku": 0.97,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.963,
"cell": 0.9857097491267068,
"valid_sudoku": 0.963,
"clues_kept": 1.0,
"nfe": 35.75
},
"entbudget_0.05": {
"board": 0.97,
"cell": 0.9853569034261318,
"valid_sudoku": 0.97,
"clues_kept": 1.0,
"nfe": 24.75
},
"entbudget_0.2": {
"board": 0.966,
"cell": 0.9850040577255567,
"valid_sudoku": 0.966,
"clues_kept": 1.0,
"nfe": 26.0
},
"entbudget_0.5": {
"board": 0.961,
"cell": 0.9856391799865919,
"valid_sudoku": 0.961,
"clues_kept": 1.0,
"nfe": 22.75
},
"entbudget_1.0": {
"board": 0.956,
"cell": 0.9811051127342013,
"valid_sudoku": 0.956,
"clues_kept": 1.0,
"nfe": 14.5
},
"entbudget_2.0": {
"board": 0.958,
"cell": 0.982798772096962,
"valid_sudoku": 0.958,
"clues_kept": 1.0,
"nfe": 8.0
},
"conf_0.9": {
"board": 0.959,
"cell": 0.9802229984827635,
"valid_sudoku": 0.959,
"clues_kept": 1.0,
"nfe": 14.0
},
"conf_0.99": {
"board": 0.973,
"cell": 0.9887618644366818,
"valid_sudoku": 0.973,
"clues_kept": 1.0,
"nfe": 24.25
},
"conf_0.999": {
"board": 0.968,
"cell": 0.9864860096679722,
"valid_sudoku": 0.968,
"clues_kept": 1.0,
"nfe": 32.25
}
},
"best_board": 0.973
}