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