GOVINDFROM's picture
Recurrent masked diffusion: results, code, card
ae02605 verified
Raw History Blame Contribute Delete
3.07 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": 1,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/abl-nodeepsup-s1"
},
"decode": {
"fixedK_1": {
"board": 0.935,
"cell": 0.9744892558484175,
"valid_sudoku": 0.935,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.94,
"cell": 0.9745951095585901,
"valid_sudoku": 0.94,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.956,
"cell": 0.9811756818743164,
"valid_sudoku": 0.956,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.968,
"cell": 0.9852157651459017,
"valid_sudoku": 0.968,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.972,
"cell": 0.9868917822236336,
"valid_sudoku": 0.972,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.975,
"cell": 0.9883031650259342,
"valid_sudoku": 0.975,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.976,
"cell": 0.9873328393493526,
"valid_sudoku": 0.976,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.982,
"cell": 0.9916904837514555,
"valid_sudoku": 0.982,
"clues_kept": 1.0,
"nfe": 31.25
},
"entbudget_0.05": {
"board": 0.98,
"cell": 0.9899968243886949,
"valid_sudoku": 0.98,
"clues_kept": 1.0,
"nfe": 22.5
},
"entbudget_0.2": {
"board": 0.979,
"cell": 0.9905966620796726,
"valid_sudoku": 0.979,
"clues_kept": 1.0,
"nfe": 19.5
},
"entbudget_0.5": {
"board": 0.975,
"cell": 0.9885148724462792,
"valid_sudoku": 0.975,
"clues_kept": 1.0,
"nfe": 12.5
},
"entbudget_1.0": {
"board": 0.961,
"cell": 0.9830634063723933,
"valid_sudoku": 0.961,
"clues_kept": 1.0,
"nfe": 8.0
},
"entbudget_2.0": {
"board": 0.949,
"cell": 0.9780353551391976,
"valid_sudoku": 0.949,
"clues_kept": 1.0,
"nfe": 7.0
},
"conf_0.9": {
"board": 0.959,
"cell": 0.9837338132034861,
"valid_sudoku": 0.959,
"clues_kept": 1.0,
"nfe": 8.5
},
"conf_0.99": {
"board": 0.974,
"cell": 0.989044140997142,
"valid_sudoku": 0.974,
"clues_kept": 1.0,
"nfe": 20.5
},
"conf_0.999": {
"board": 0.979,
"cell": 0.9893969866977171,
"valid_sudoku": 0.979,
"clues_kept": 1.0,
"nfe": 30.0
}
},
"best_board": 0.982
}