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

do not enable IDM for expert data

Browse Source
This commit is contained in:
ebuehrle
2022-09-22 11:15:49 +02:00
parent 740e0ea9f4
commit 5fb358d725
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
eval_experiments.py
View File

@@ -11,8 +11,10 @@ def main(method:str='expert', folder:str=None, locations=[(0,0)], skip_running=F
exclude_keys_from_policy_kwargs = {'learning_rate', 'learning_rate_decay', 'clip_ratio', 'iterations_per_epoch', 'option'} exclude_keys_from_policy_kwargs = {'learning_rate', 'learning_rate_decay', 'clip_ratio', 'iterations_per_epoch', 'option'}
policy_kwargs = {} policy_kwargs = {}
if method in ['expert', 'expert_agent', 'idm']: if method in ['expert', 'expert_agent']:
env, env_kwargs ='NRasterizedRouteIncrementingAgent', {} env, env_kwargs ='NRasterizedRouteIncrementingAgent', {}
elif method in ['idm']:
env, env_kwargs ='NRasterizedRouteIncrementingAgent', {'use_idm':True}
elif method in ['bc','gail']: elif method in ['bc','gail']:
env='NormalizedContinuousEvalEnv' env='NormalizedContinuousEvalEnv'
env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000, 'use_idm':True} env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000, 'use_idm':True}