File size: 822 Bytes
1d83a22
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
{
    "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-128-64-24",
        "--arch-mlp-top": "1024-256-64-1",
        "--arch-sparse-feature-size": "24",
        "--inference-only": "",
        "--load-model": "@cwd@/dlrm_RMC3_cat.pt",
        "--data-generation": "dataset",
        "--data-set": "kaggle",
        "--processed-data-file": "@cwd@/../../datasets/media/criteo_kaggle/kaggleAdDisplayChallenge21Tables_processed.npz",
        "--raw-data-file": "@cwd@/../../datasets/media/criteo_kaggle/train.txt",
        "--arch-interaction-op": "cat"
    }
}