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