GOVINDFROM's picture
Recurrent masked diffusion: results, code, card
ae02605 verified
Raw History Blame Contribute Delete
3.09 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": 0,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/base-directrec-s0"
},
"decode": {
"fixedK_1": {
"board": 0.358,
"cell": 0.8523870011643908,
"valid_sudoku": 0.358,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.077,
"cell": 0.7139127059736777,
"valid_sudoku": 0.077,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.0,
"cell": 0.5832715853357326,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.0,
"cell": 0.574468085106383,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.0,
"cell": 0.5897286616562577,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.0,
"cell": 0.5926043541159451,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.0,
"cell": 0.5926396386860026,
"valid_sudoku": 0.0,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.129,
"cell": 0.760117850463992,
"valid_sudoku": 0.129,
"clues_kept": 1.0,
"nfe": 57.0
},
"entbudget_0.05": {
"board": 0.318,
"cell": 0.8145795843477647,
"valid_sudoku": 0.318,
"clues_kept": 1.0,
"nfe": 51.25
},
"entbudget_0.2": {
"board": 0.381,
"cell": 0.8257471507709678,
"valid_sudoku": 0.381,
"clues_kept": 1.0,
"nfe": 41.75
},
"entbudget_0.5": {
"board": 0.39,
"cell": 0.8347623584206626,
"valid_sudoku": 0.39,
"clues_kept": 1.0,
"nfe": 32.5
},
"entbudget_1.0": {
"board": 0.381,
"cell": 0.8299812991778696,
"valid_sudoku": 0.381,
"clues_kept": 1.0,
"nfe": 26.5
},
"entbudget_2.0": {
"board": 0.38,
"cell": 0.8337391058889947,
"valid_sudoku": 0.38,
"clues_kept": 1.0,
"nfe": 20.0
},
"conf_0.9": {
"board": 0.401,
"cell": 0.8617550545146607,
"valid_sudoku": 0.401,
"clues_kept": 1.0,
"nfe": 25.75
},
"conf_0.99": {
"board": 0.409,
"cell": 0.8534278959810875,
"valid_sudoku": 0.409,
"clues_kept": 1.0,
"nfe": 44.25
},
"conf_0.999": {
"board": 0.329,
"cell": 0.8302282911682721,
"valid_sudoku": 0.329,
"clues_kept": 1.0,
"nfe": 54.0
}
},
"best_board": 0.409
}