Skip to content

GitLab

Commit 44550d37 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Fix script name for DLRM model 

PiperOrigin-RevId: 379909569
parent 85a48cba
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/ranking/README.md
View file @ 44550d37
...@@ -154,7 +154,7 @@ Training on GPUs are similar to TPU training. Only distribution strategy needs ...@@ -154,7 +154,7 @@ Training on GPUs are similar to TPU training. Only distribution strategy needs
to be updated and number of GPUs provided (for 4 GPUs): to be updated and number of GPUs provided (for 4 GPUs):
```shell ```shell
python3 official/recommendation/ranking/main.py --mode=train_and_eval \ python3 official/recommendation/ranking/train.py --mode=train_and_eval \
--model_dir=${BUCKET_NAME}/model_dirs/${EXPERIMENT_NAME} --params_override=" --model_dir=${BUCKET_NAME}/model_dirs/${EXPERIMENT_NAME} --params_override="
runtime: runtime:
distribution_strategy: 'mirrored' distribution_strategy: 'mirrored'
......
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