huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optionally save videos of policy evaluations

Browse Source
This commit is contained in:
ebuehrle
2022-03-04 15:58:18 +01:00
parent a9feec4f38
commit 1e70303c57
9 changed files with 95 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
generate_videos.sh Executable file
View File

@@ -0,0 +1,18 @@
# can add --skip_running if you've already run the saved policies through the test environments and have appropriate
# metrics in the out folder. Doing so will generate average metrics quickly.
# Experiment A
python -m eval_experiments
python -m eval_experiments --method idm --save_videos
python -m eval_experiments --method bc --folder='test_policies/bc/expA' --save_videos
python -m eval_experiments --method gail --folder='test_policies/gail/expA' --save_videos
python -m eval_experiments --method hail --folder='test_policies/hail/expA' --save_videos
python -m eval_experiments --method shail --folder='test_policies/shail/expA' --save_videos
# Experiment B
python -m eval_experiments --locations='[(0,4)]'
python -m eval_experiments --method idm --locations='[(0,4)]' --save_videos
python -m eval_experiments --method bc --folder='test_policies/bc/expB' --locations='[(0,4)]' --save_videos
python -m eval_experiments --method gail --folder='test_policies/gail/expB' --locations='[(0,4)]' --save_videos
python -m eval_experiments --method hail --folder='test_policies/hail/expB' --locations='[(0,4)]' --save_videos
python -m eval_experiments --method shail --folder='test_policies/shail/expB' --locations='[(0,4)]' --save_videos