From a363f162870979d51188c7b5549d66e2f34ccd04 Mon Sep 17 00:00:00 2001 From: alexlee-gk <alexleegk@gmail.com> Date: Sat, 7 Apr 2018 20:22:37 -0700 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index ea870989..5a368a30 100644 --- a/README.md +++ b/README.md @@ -44,7 +44,7 @@ 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 \ + --model savp --model_hparams_dict hparams/bair_action_free/ours_savp/model_hparams.json \ --logs_dir logs/bair_action_free ``` - 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. -- GitLab