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