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": 1,
"eval_bs": 250,
"ckpt_every": 10000,
"out": "/scratch/zt1/project/msml612/user/govind02/pcmd/runs/fix-directall-s1"
},
"decode": {
"fixedK_1": {
"board": 0.094,
"cell": 0.5431353868953107,
"valid_sudoku": 0.094,
"clues_kept": 1.0,
"nfe": 1.0
},
"fixedK_2": {
"board": 0.084,
"cell": 0.5583606788751279,
"valid_sudoku": 0.084,
"clues_kept": 1.0,
"nfe": 2.0
},
"fixedK_4": {
"board": 0.03,
"cell": 0.5073568328569916,
"valid_sudoku": 0.03,
"clues_kept": 1.0,
"nfe": 4.0
},
"fixedK_8": {
"board": 0.007,
"cell": 0.5128965103560214,
"valid_sudoku": 0.007,
"clues_kept": 1.0,
"nfe": 8.0
},
"fixedK_16": {
"board": 0.01,
"cell": 0.5899050845065453,
"valid_sudoku": 0.01,
"clues_kept": 1.0,
"nfe": 16.0
},
"fixedK_32": {
"board": 0.007,
"cell": 0.6276948590381426,
"valid_sudoku": 0.007,
"clues_kept": 1.0,
"nfe": 32.0
},
"fixedK_61": {
"board": 0.014,
"cell": 0.6588158498288699,
"valid_sudoku": 0.014,
"clues_kept": 1.0,
"nfe": 59.5
},
"entbudget_0.01": {
"board": 0.088,
"cell": 0.715923926466956,
"valid_sudoku": 0.088,
"clues_kept": 1.0,
"nfe": 58.25
},
"entbudget_0.05": {
"board": 0.138,
"cell": 0.7457393881655552,
"valid_sudoku": 0.138,
"clues_kept": 1.0,
"nfe": 57.75
},
"entbudget_0.2": {
"board": 0.187,
"cell": 0.7710913517518789,
"valid_sudoku": 0.187,
"clues_kept": 1.0,
"nfe": 58.0
},
"entbudget_0.5": {
"board": 0.212,
"cell": 0.7857344483257471,
"valid_sudoku": 0.212,
"clues_kept": 1.0,
"nfe": 57.0
},
"entbudget_1.0": {
"board": 0.214,
"cell": 0.788504287075262,
"valid_sudoku": 0.214,
"clues_kept": 1.0,
"nfe": 57.0
},
"entbudget_2.0": {
"board": 0.189,
"cell": 0.8019829928372323,
"valid_sudoku": 0.189,
"clues_kept": 1.0,
"nfe": 57.5
},
"conf_0.9": {
"board": 0.194,
"cell": 0.7980840478458769,
"valid_sudoku": 0.194,
"clues_kept": 1.0,
"nfe": 57.75
},
"conf_0.99": {
"board": 0.213,
"cell": 0.7785893228891005,
"valid_sudoku": 0.213,
"clues_kept": 1.0,
"nfe": 58.0
},
"conf_0.999": {
"board": 0.149,
"cell": 0.7357009279841925,
"valid_sudoku": 0.149,
"clues_kept": 1.0,
"nfe": 58.25
}
},
"best_board": 0.214
}