temp magic
This commit is contained in:
parent
5a1e29f710
commit
b3c17f803f
1 changed files with 2 additions and 0 deletions
|
|
@ -103,6 +103,8 @@ def main(dict_cfg: DictConfig) -> None:
|
||||||
obs_processor = create_obs_processor(
|
obs_processor = create_obs_processor(
|
||||||
bounds_dict=training.obs_bounds.to_bounds_dict(),
|
bounds_dict=training.obs_bounds.to_bounds_dict(),
|
||||||
padding_masks=padding_masks,
|
padding_masks=padding_masks,
|
||||||
|
num_arms=5,
|
||||||
|
needed_copies=5,
|
||||||
)
|
)
|
||||||
|
|
||||||
env = BrittleStarJaxEnvWrapper(
|
env = BrittleStarJaxEnvWrapper(
|
||||||
|
|
|
||||||
Reference in a new issue