Skip to content

Commit

Permalink
eval command added
Browse files Browse the repository at this point in the history
  • Loading branch information
AlmondGod committed Oct 14, 2024
1 parent c65b5b3 commit afc7949
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion models/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,4 +18,14 @@ To train ACT:
--seed 0


To evaluate the policy, run the same command but add ``--eval``. This loads the best validation checkpoint.
To evaluate the policy, run the same command but add ``--eval``. This loads the best validation checkpoint.


to eval:

mjpython model_eval.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 \
--num_epochs 2000 --lr 1e-5 \
--seed 0 --eval

0 comments on commit afc7949

Please sign in to comment.