| { | |
| "framework": "pytorch", | |
| "classifier": "@cwd@/dlrm_s_pytorch.py", | |
| "input-data-info": "autogenerated", | |
| "input-download": "datasets/media/criteo_kaggle", | |
| "flags": { | |
| "--minibatch": 128, | |
| "--num-batches": 1, | |
| "--data-generation": "random", | |
| "--arch-mlp-bot": "13-512-256-64-16", | |
| "--arch-mlp-top": "512-256-1", | |
| "--arch-sparse-feature-size": "16", | |
| "--inference-only": "", | |
| "--load-model": "@cwd@/dlrm_RMC2_kaggle_cat.pt", | |
| "--data-generation": "dataset", | |
| "--data-set": "kaggle", | |
| "--processed-data-file": "@cwd@/../../datasets/media/criteo_kaggle/kaggleAdDisplayChallenge26Tables_processed.npz", | |
| "--raw-data-file": "@cwd@/../../datasets/media/criteo_kaggle/train.txt", | |
| "--arch-interaction-op": "cat" | |
| } | |
| } | |