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

Add train_all.sh script (untested).

parent 9f296f63
No related branches found
No related tags found
No related merge requests found
# BAIR action-free robot pushing dataset
for model in \
ours_deterministic_l1 \
ours_deterministic_l2 \
ours_vae_l1 \
ours_vae_l2 \
ours_gan \
ours_savp \
; do
CUDA_VISIBLE_DEVICES=0 python scripts/train.py --input_dir data/bair --dataset bair --model savp --model_hparams_dict hparams/bair_action_free/${model}.json --logs_dir logs/bair_action_free
done
# KTH human actions dataset
for model in \
ours_deterministic_l1 \
ours_deterministic_l2 \
ours_vae_l1 \
ours_gan \
ours_savp \
; do
CUDA_VISIBLE_DEVICES=0 python scripts/train.py --input_dir data/kth--dataset kth --model savp --model_hparams_dict hparams/kth/${model}.json --logs_dir logs/kth
done
# BAIR action-conditioned robot pushing dataset
for model in \
ours_deterministic_l1 \
ours_deterministic_l2 \
ours_vae_l1 \
ours_vae_l2 \
ours_gan \
ours_savp \
; do
CUDA_VISIBLE_DEVICES=0 python scripts/train.py --input_dir data/bair --dataset bair --dataset_hparams use_state=True --model savp --model_hparams_dict hparams/bair/${model}.json --logs_dir logs/bair
done
for model in \
sna_l1 \
sna_l2 \
; do
CUDA_VISIBLE_DEVICES=0 python scripts/train.py --input_dir data/bair --dataset bair --dataset_hparams use_state=True --model sna --model_hparams_dict hparams/bair/${model}.json --logs_dir logs/bair
done
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment