diff --git a/README.md b/README.md index 9f90f3e..2a34515 100644 --- a/README.md +++ b/README.md @@ -74,7 +74,7 @@ To train ACT: --task_name sim_transfer_cube_scripted \ --ckpt_dir \ --policy_class ACT --kl_weight 10 --chunk_size 100 --hidden_dim 512 --batch_size 8 --dim_feedforward 3200 \ - --num_epochs 2000 --lr 1e-5 \temporal_agg + --num_epochs 2000 --lr 1e-5 \ --seed 0