working with obs=dict(state=state, img=img)

2026-09-09 11:20:39 +08:00 · 2021-02-18 11:57:29 +08:00
parent 8416d6e386
commit b1eb2efb77
10 changed files with 192 additions and 96 deletions
@@ -1,9 +1,5 @@
-CUDA_VISIBLE_DEVICES=5 python train.py \
-    --domain_name cartpole \
-    --task_name swingup \
-    --encoder_type pixel \
-    --action_repeat 8 \
-    --save_tb --pre_transform_image_size 100 --image_size 84 \
-    --work_dir ./tmp/cartpole \
-    --agent curl_sac --frame_stack 3 \
-    --seed -1 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 10000 --batch_size 128 --num_train_steps 1000000 
+#!/bin/bash
+CUDA_VISIBLE_DEVICES=1 python \
+    -m pdb -c continue \
+    train.py \
+        --save_tb