File size: 821 Bytes
0a6b525 |
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-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"
}
}
|