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
......@@ -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):
```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="
runtime:
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