Skip to content

GitLab

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

Update README.md

parent 9c8a2a14
Pipeline #3003 failed with stages
in 0 seconds
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ 8d6336ba
...@@ -121,7 +121,7 @@ python dlrm_s_pytorch.py --mini-batch-size=2 --data-size=6 --debug-mode ...@@ -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] ./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"> <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 ...@@ -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"] ./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) - 对应的预训练模型可从以下链接下载：[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