Skip to content
Snippets Groups Projects
Commit 72f1f337 authored by Alex Lee's avatar Alex Lee
Browse files

Merge branch 'master' of github.com:alexlee-gk/video_prediction

parents 6d0e1d93 a2c9d40b
No related branches found
No related tags found
No related merge requests found
...@@ -44,8 +44,8 @@ bash data/download_and_preprocess_dataset.sh bair ...@@ -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): - Train a model (e.g. our SAVP model on the BAIR action-free robot pushing dataset):
```bash ```bash
CUDA_VISIBLE_DEVICES=0 python scripts/train.py --input_dir data/bair --dataset bair \ 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 --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. - 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=""`. - 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=""`.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment