imitation_nonconvex: Run for longer.

brunompacheco · Nov 25, 2018 · 5e40c02 · 5e40c02
1 parent 54eec96
commit 5e40c02
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/imitation_nonconvex/run-pendulum-cartpole.sh b/imitation_nonconvex/run-pendulum-cartpole.sh
@@ -1,11 +1,13 @@
 #!/bin/bash
 
+N_EPOCH=10000
+
 args_all_modes() {
-    echo --mode nn --no-cuda $*
-    echo --mode sysid --no-cuda $*
-    echo --mode empc --no-cuda --learn_cost $*
-    echo --mode empc --no-cuda --learn_dx $*
-    echo --mode empc --no-cuda --learn_cost --learn_dx $*
+    echo --n_epoch $N_EPOCH --mode nn --no-cuda $*
+    echo --n_epoch $N_EPOCH --mode sysid --no-cuda $*
+    echo --n_epoch $N_EPOCH --mode empc --no-cuda --learn_cost $*
+    echo --n_epoch $N_EPOCH --mode empc --no-cuda --learn_dx $*
+    echo --n_epoch $N_EPOCH --mode empc --no-cuda --learn_cost --learn_dx $*
 }
 
 args_all_sizes() {
@@ -31,7 +33,7 @@ export -f run_single
 
 export OMP_NUM_THREADS=1
 
-MAX_PROCS=34
+MAX_PROCS=16
 args_all_seeds ./data/pendulum.pkl | parallel --no-notice --max-procs $MAX_PROCS run_single &
 args_all_seeds ./data/cartpole.pkl | parallel --no-notice --max-procs $MAX_PROCS run_single &
 wait