GOVINDFROM's picture
Recurrent masked diffusion: results, code, card
ae02605 verified
Raw History Blame Contribute Delete
3.13 kB
{
"args": {
"mode": "mdlm",
"all_cells": true,
"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": 110000,
"lr": 0.0003,
"warmup": 3000,
"seed": 0,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/fix-directall-s0"
},
"decode": {
"fixedK_1": {
"board": 0.105,
"cell": 0.6825270809075191,
"valid_sudoku": 0.105,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.154,
"cell": 0.7321371864083837,
"valid_sudoku": 0.154,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.147,
"cell": 0.7680568787269327,
"valid_sudoku": 0.147,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.259,
"cell": 0.8128506404149466,
"valid_sudoku": 0.259,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.318,
"cell": 0.8437775660703575,
"valid_sudoku": 0.318,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.398,
"cell": 0.8605553791327053,
"valid_sudoku": 0.398,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.452,
"cell": 0.881620267457041,
"valid_sudoku": 0.452,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.501,
"cell": 0.893017183585618,
"valid_sudoku": 0.501,
"clues_kept": 1.0,
"nfe": 58.5
},
"entbudget_0.05": {
"board": 0.525,
"cell": 0.9052609293955753,
"valid_sudoku": 0.525,
"clues_kept": 1.0,
"nfe": 57.75
},
"entbudget_0.2": {
"board": 0.549,
"cell": 0.9146819096009315,
"valid_sudoku": 0.549,
"clues_kept": 1.0,
"nfe": 56.5
},
"entbudget_0.5": {
"board": 0.493,
"cell": 0.9018383260999965,
"valid_sudoku": 0.493,
"clues_kept": 1.0,
"nfe": 55.0
},
"entbudget_1.0": {
"board": 0.432,
"cell": 0.905578490526093,
"valid_sudoku": 0.432,
"clues_kept": 1.0,
"nfe": 49.0
},
"entbudget_2.0": {
"board": 0.322,
"cell": 0.8803676652199993,
"valid_sudoku": 0.322,
"clues_kept": 1.0,
"nfe": 32.75
},
"conf_0.9": {
"board": 0.33,
"cell": 0.9106241840443174,
"valid_sudoku": 0.33,
"clues_kept": 1.0,
"nfe": 54.75
},
"conf_0.99": {
"board": 0.551,
"cell": 0.9083306869905791,
"valid_sudoku": 0.551,
"clues_kept": 1.0,
"nfe": 58.0
},
"conf_0.999": {
"board": 0.532,
"cell": 0.906954588758336,
"valid_sudoku": 0.532,
"clues_kept": 1.0,
"nfe": 58.75
}
},
"best_board": 0.551
}