Skip to content
Snippets Groups Projects
Unverified Commit a363f162 authored by alexlee-gk's avatar alexlee-gk Committed by GitHub
Browse files

Update README.md

parent ba30b06d
No related branches found
No related tags found
No related merge requests found
......@@ -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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment