diff --git a/hparams/bair/ours_vae_gan/model_hparams.json b/hparams/bair/ours_savp/model_hparams.json similarity index 100% rename from hparams/bair/ours_vae_gan/model_hparams.json rename to hparams/bair/ours_savp/model_hparams.json diff --git a/hparams/bair_action_free/ours_vae_gan/model_hparams.json b/hparams/bair_action_free/ours_savp/model_hparams.json similarity index 100% rename from hparams/bair_action_free/ours_vae_gan/model_hparams.json rename to hparams/bair_action_free/ours_savp/model_hparams.json diff --git a/hparams/kth/ours_vae_gan/model_hparams.json b/hparams/kth/ours_savp/model_hparams.json similarity index 100% rename from hparams/kth/ours_vae_gan/model_hparams.json rename to hparams/kth/ours_savp/model_hparams.json diff --git a/scripts/train_all.sh b/scripts/train_all.sh new file mode 100644 index 0000000000000000000000000000000000000000..82ce0b8b63c7b98a43ab7e26634aeb01a76f02c8 --- /dev/null +++ b/scripts/train_all.sh @@ -0,0 +1,40 @@ +# 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