From 541223fc816d52e2187d52be6cc06525ac41f02e Mon Sep 17 00:00:00 2001 From: Alex Lee <alexleegk@gmail.com> Date: Sat, 7 Apr 2018 20:19:17 -0700 Subject: [PATCH] Add train_all.sh script (untested). --- .../model_hparams.json | 0 .../model_hparams.json | 0 .../model_hparams.json | 0 scripts/train_all.sh | 40 +++++++++++++++++++ 4 files changed, 40 insertions(+) rename hparams/bair/{ours_vae_gan => ours_savp}/model_hparams.json (100%) rename hparams/bair_action_free/{ours_vae_gan => ours_savp}/model_hparams.json (100%) rename hparams/kth/{ours_vae_gan => ours_savp}/model_hparams.json (100%) create mode 100644 scripts/train_all.sh 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 00000000..82ce0b8b --- /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 -- GitLab