{
  "dataset": "AL-GR/Item-EMB",
  "cache_path": "data/item_emb.npy",
  "npz_path": "data/reconstruct_data_mask.npz",
  "max_items": 0,
  "output_dir": "ckpts/forge_rqvae",
  "codebook_num": 3,
  "codebook_size": 8192,
  "codebook_dim": 64,
  "batch_size": 2048,
  "epochs": 150,
  "warmup_epochs": 40,
  "lr": 0.0002,
  "min_lr": 1e-07,
  "weight_decay": 0.0001,
  "seed": 0,
  "save_every": 25,
  "eval_every": 5,
  "bf16": true,
  "num_workers": 4,
  "hub_model_id": "arusl/forge-rqvae-3x8192",
  "push_dataset_id": "arusl/forge-rqvae-sids",
  "trackio_space": "arusl/mlintern-94e805f5",
  "trackio_project": "forge-rqvae",
  "run_name": "rqvae_3x8192_lr2e-4_bs2048_ep150"
}