diff --git a/README.md b/README.md
index ea870989e15688133aa7fe80dda9aae5d08f73ec..abbd9e210b2dbb880576dfcc31f887d02d7358f7 100644
--- a/README.md
+++ b/README.md
@@ -44,8 +44,8 @@ bash data/download_and_preprocess_dataset.sh bair
 - Train a model (e.g. our SAVP model on the BAIR action-free robot pushing dataset):
 ```bash
 CUDA_VISIBLE_DEVICES=0 python scripts/train.py --input_dir data/bair --dataset bair \
-  --model savp --model_hparams_dict hparams/bair_action_free/ours_savp.json \
-  --logs_dir logs/bair_action_free
+  --model savp --model_hparams_dict hparams/bair_action_free/ours_savp/model_hparams.json \
+  --output_dir logs/bair_action_free/${model}
 ```
 - To view training and validation information (e.g. loss plots, GIFs of predictions), run `tensorboard --logdir logs/bair_action_free --port 6006` and open http://localhost:6006.
 - For multi-GPU training, set `CUDA_VISIBLE_DEVICES` to a comma-separated list of devices, e.g. `CUDA_VISIBLE_DEVICES=0,1,2,3`. To use the CPU, set `CUDA_VISIBLE_DEVICES=""`.