Fix script name for DLRM model

PiperOrigin-RevId: 379909569
tensorflow · Jun 17, 2021 · 66264b2 · 66264b2
1 parent 359a51b
commit 66264b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/official/recommendation/ranking/README.md b/official/recommendation/ranking/README.md
@@ -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'