Poker-Hand / checkpoints /config.json
fjrmhri's picture
Upload model artifacts (checkpoints/state_dict) from Colab
903692d verified
{
"SEED": 42,
"DETERMINISTIC": false,
"SPLIT_MODE": "standard",
"TRAIN_RATIO": 0.8,
"VAL_RATIO": 0.1,
"TEST_RATIO": 0.1,
"EMB_DIM": 128,
"N_HEADS": 4,
"N_LAYERS": 4,
"MLP_HIDDEN": 256,
"DROPOUT": 0.1,
"NUM_CLASSES": 10,
"NUM_STAGES": 3,
"NUM_RANKS": 13,
"NUM_SUITS": 4,
"IMBALANCE_MODE": "sampler",
"FOCAL_GAMMA": 2.0,
"FOCAL_ALPHA": null,
"CLASS_WEIGHT_POWER": 0.5,
"CLASS_WEIGHT_CAP": 10.0,
"BATCH_SIZE": 2048,
"NUM_WORKERS": 2,
"EPOCHS": 30,
"PATIENCE": 5,
"MIN_DELTA": 0.0001,
"LR": 0.001,
"WEIGHT_DECAY": 0.001,
"LR_MIN": 1e-05,
"GRAD_CLIP": 1.0,
"USE_STRENGTH_HEAD": true,
"STRENGTH_LOSS_WEIGHT": 0.5,
"CACHE_DIR": "cache",
"USE_CACHE": true,
"CKPT_DIR": "checkpoints"
}