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

adding idm override option flag, set to true. current running experiments for gail and shail experiment A to see how different times are. Since were on cpus on the cluster, guessing it will be 10x

Browse Source
This commit is contained in:
Arec Jamgochian
2022-08-07 16:33:09 -07:00
parent 9c9ee8f21b
commit 3a09a6eb7d
5 changed files with 21 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
gail-experiment.py
View File

@@ -53,6 +53,7 @@ def training_function(config):
),
check_collisions=True,
stop_on_collision=config['trainenv']['stop_on_collision'],
use_idm=config['trainenv']['use_idm'],
), collision_distance=6, collision_penalty=100),
lambda obs: (obs - obs_min) / (obs_max - obs_min + 1e-10)
)) for _ in range(60)]
@@ -67,6 +68,7 @@ def training_function(config):
),
check_collisions=True,
stop_on_collision=config['trainenv']['stop_on_collision'],
use_idm=config['trainenv']['use_idm'],
track=track,
), collision_distance=6, collision_penalty=100),
lambda obs: (obs - obs_min) / (obs_max - obs_min + 1e-10)
@@ -169,7 +171,8 @@ if __name__ == '__main__':
config={
'experiment': args.train,
'trainenv': {
'stop_on_collision': False,
'stop_on_collision': False,
'use_idm': True,
},
'policy': {
'learning_rate': 3e-4,