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

Make envs independent in BC experiment

Browse Source
This commit is contained in:
ebuehrle
2022-02-27 22:53:12 +01:00
parent 7f1557144c
commit 25e4d5679f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bc-experiment.py
View File

@@ -44,7 +44,7 @@ def training_function(config):
np.random.seed(config['seed']) np.random.seed(config['seed'])
torch.manual_seed(config['seed']) torch.manual_seed(config['seed'])
env = Setobs(TransformObservation(CollisionPenaltyWrapper( envs = [Setobs(TransformObservation(CollisionPenaltyWrapper(
IntersimpleLidarFlatRandom( IntersimpleLidarFlatRandom(
n_rays=5, n_rays=5,
reward=functools.partial( reward=functools.partial(
@@ -55,8 +55,8 @@ def training_function(config):
stop_on_collision=config['trainenv']['stop_on_collision'], stop_on_collision=config['trainenv']['stop_on_collision'],
), collision_distance=6, collision_penalty=100), ), collision_distance=6, collision_penalty=100),
lambda obs: (obs - obs_min) / (obs_max - obs_min + 1e-10) lambda obs: (obs - obs_min) / (obs_max - obs_min + 1e-10)
)) )) for _ in range(60)]
env_fn = lambda i: env env_fn = lambda i: envs[i]
# load expert data # load expert data