"tests/experimental/vscode:/vscode.git/clone" did not exist on "f4fcee7e8a74660bb34114cff710c769167c936b"
Commit 8d6336ba authored by xinghao's avatar xinghao
Browse files

Update README.md

parent 9c8a2a14
Pipeline #3003 failed with stages
in 0 seconds
......@@ -121,7 +121,7 @@ python dlrm_s_pytorch.py --mini-batch-size=2 --data-size=6 --debug-mode
./bench/dlrm_s_criteo_kaggle.sh [--test-freq=1024]
```
若要启用gpu,添加参数 `--use-gpu`若要启用纯推理模,添加参数 `--inference-only` 并使用参数 `--load-model`指定权重文件
若要启用gpu,添加参数 `--use-gpu`若要启用纯推理模,添加参数 `--inference-only` 并使用参数 `--load-model`指定权重文件
<img src="./kaggle_dac_loss_accuracy_plots.png" width="900" height="320">
......@@ -139,7 +139,7 @@ python dlrm_s_pytorch.py --mini-batch-size=2 --data-size=6 --debug-mode
./bench/dlrm_s_criteo_terabyte.sh ["--test-freq=10240 --memory-map --data-sub-sample-rate=0.875"]
```
​ 若要启用gpu,添加参数 `--use-gpu`若要启用纯推理模,添加参数 `--inference-only` 并使用参数 `--load-model`指定权重文件
​ 若要启用gpu,添加参数 `--use-gpu`若要启用纯推理模,添加参数 `--inference-only` 并使用参数 `--load-model`指定权重文件
- 对应的预训练模型可从以下链接下载:[dlrm_emb64_subsample0.875_maxindrange10M_pretrained.pt](https://dlrm.s3-us-west-1.amazonaws.com/models/tb0875_10M.pt)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment