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": 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": 32,
"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-deep32L-s0"
},
"decode": {
"fixedK_1": {
"board": 0.702,
"cell": 0.9354115945097209,
"valid_sudoku": 0.702,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.832,
"cell": 0.9493666419674677,
"valid_sudoku": 0.832,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.865,
"cell": 0.9574115239405808,
"valid_sudoku": 0.865,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.906,
"cell": 0.9639391694012208,
"valid_sudoku": 0.906,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.911,
"cell": 0.9625807134540065,
"valid_sudoku": 0.911,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.918,
"cell": 0.9660386013196429,
"valid_sudoku": 0.918,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.928,
"cell": 0.9694082777601355,
"valid_sudoku": 0.928,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.945,
"cell": 0.9751596626795103,
"valid_sudoku": 0.945,
"clues_kept": 1.0,
"nfe": 45.25
},
"entbudget_0.05": {
"board": 0.944,
"cell": 0.9752478741046541,
"valid_sudoku": 0.944,
"clues_kept": 1.0,
"nfe": 33.25
},
"entbudget_0.2": {
"board": 0.933,
"cell": 0.9735189301718359,
"valid_sudoku": 0.933,
"clues_kept": 1.0,
"nfe": 24.75
},
"entbudget_0.5": {
"board": 0.922,
"cell": 0.9706961645672347,
"valid_sudoku": 0.922,
"clues_kept": 1.0,
"nfe": 16.25
},
"entbudget_1.0": {
"board": 0.873,
"cell": 0.9671147807063971,
"valid_sudoku": 0.873,
"clues_kept": 1.0,
"nfe": 10.25
},
"entbudget_2.0": {
"board": 0.823,
"cell": 0.9554532303023887,
"valid_sudoku": 0.823,
"clues_kept": 1.0,
"nfe": 8.0
},
"conf_0.9": {
"board": 0.837,
"cell": 0.9656328287639815,
"valid_sudoku": 0.837,
"clues_kept": 1.0,
"nfe": 14.75
},
"conf_0.99": {
"board": 0.931,
"cell": 0.9755477929501429,
"valid_sudoku": 0.931,
"clues_kept": 1.0,
"nfe": 27.5
},
"conf_0.999": {
"board": 0.942,
"cell": 0.9744186867083025,
"valid_sudoku": 0.942,
"clues_kept": 1.0,
"nfe": 41.0
}
},
"best_board": 0.945
}