checkpoints/config.json

{
  "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"
}