search

jurgisp / pydreamer

PyTorch implementation of DreamerV2 model-based RL algorithm
MIT License
209 stars 48 forks source link

procgen env #2

Closed hlsfin closed 2 years ago

hlsfin commented 2 years ago

sorry I'm sort of new in this field, Can I run this on procgen enviroment?

jurgisp commented 2 years ago

Yes, you should be able to! There are special constructors for some environments, but by default, pydreamer tries to create a gym environment with given env_id here, which should be enough for Procgen.

So something like this should work:

pip pinstall procgen
python train.py --configs defaults --env_id procgen:procgen-coinrun-v0 --action_dim 15
hlsfin commented 2 years ago

Thank you so much!