google / trajax

Apache License 2.0
186 stars 23 forks source link

Incorporating `frozendict` into default CEM hyperparameters function #12

Open kwesiRutledge opened 11 months ago

kwesiRutledge commented 11 months ago

Incorporated Small Changes (Using FrozenDict + Providing Default empty dictionaries to costs vmap) That Seem To Make Random Shooting Happy

This is following up on #3 . It seems that after incorporating the frozendict into the default_cem_hyperparams() function and adding some small changes to the internals of random_shooting, the random_shooting function no longer throws errors when using acrobot (or similar systems).