Integrate options env and policy

2022-02-15 18:36:53 +01:00
parent 072c0ff417
commit c6a4c10605
5 changed files with 60 additions and 9 deletions
--- a/evaluate_models.sh
+++ b/evaluate_models.sh
@@ -13,4 +13,4 @@ python -m src.eval_main
 # idm
 python -m src.eval_main --method=idm

-
+python -m src.eval_main --method=ogail --policy_file='checkpoints/gail-options-setobs2.pt' --env='NormalizedOptionsEvalEnv'