batch size 8->64 lr 1e-5 -> 5e-5

jrabary · Oct 14, 2024 · 673b856 · 673b856
1 parent b8416fd
commit 673b856
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/README.md b/models/README.md
@@ -13,7 +13,7 @@ To train ACT:
     python3 imitate_episodes.py \
     --task_name sim_aloha_close_dishwasher \
     --ckpt_dir ./ckpt_dir \
-    --policy_class ACT --kl_weight 10 --chunk_size 100 --hidden_dim 512 --batch_size 8 --dim_feedforward 3200 \
+    --policy_class ACT --kl_weight 10 --chunk_size 50 --hidden_dim 512 --batch_size 8 --dim_feedforward 3200 \
     --num_epochs 2000  --lr 1e-5 \
     --seed 0