video and logging

scripts/play.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+set -e
+ulimit -S -m 55000000
+ulimit -S -v 55000000
+python=/home/wassname/anaconda/envs/diygym4/bin/python
+# DEBUG
+CUDA_VISIBLE_DEVICES=1 $python \
+    -m pdb -c continue \
+    train.py \
+        --batch_size 64 \
+        --init_steps 256 \
+        --eval_freq 64 \
+        --replay_buffer_capacity 4000 \
+        --render \
+        --load auto \
+        --num_train_steps 128

scripts/run.sh

@@ -1,6 +1,30 @@
 #!/bin/bash
 set -e
-CUDA_VISIBLE_DEVICES=1 /home/wassname/anaconda/envs/diygym4/bin/python \
+ulimit -S -m 55000000
+ulimit -S -v 55000000
+python=/home/wassname/anaconda/envs/diygym4/bin/python
+CUDA_VISIBLE_DEVICES=0 
+# DEBUG
+# $python \
+#     -m pdb -c continue \
+#     train.py \
+#         --save_tb \
+#         --save_model \
+#         --save_video \
+#         --batch_size 64 \
+#         --init_steps 64 \
+#         --eval_freq 64 \
+#         --replay_buffer_capacity 4000 \
+#         --render
+
+$python \
     -m pdb -c continue \
     train.py \
-        --save_tb --init_steps 10
+        --save_tb \
+        --save_model \
+        --save_video \
+        --batch_size 512 \
+        --init_steps 10000 \
+        --replay_buffer_capacity 40000 \
+        # --load auto
+