autonomous-learning-library icon indicating copy to clipboard operation
autonomous-learning-library copied to clipboard

Feature request: multiprocess env runner

Open syuntoku14 opened this issue 5 years ago • 2 comments

Hi Chris,

Do you plan to add an EnvRunner class supporting multiprocess?

syuntoku14 avatar Feb 04 '20 16:02 syuntoku14

Maybe. For most of the standard Gym environments, the environment isn't really the bottleneck, so it's not clear that you get much of a speedup. However, it could be useful for more computationally expensive environments. Do you have any specific use case in mind?

cpnota avatar Feb 04 '20 17:02 cpnota

I think most of the RL algorithms with pybullet envs require much training time since pybullet is computationally expensive. For example, chainerrl requires 1.5 days to train kuka-diverse-object-env with DQN.

It seems that some of the continuous control scripts in your code require more than 5 hours to train, so I am implementing a simple multiprocess runner to reduce the training time. I tested it by training kuka-diverse-object-env with 30 CPUs. It saved a lot of time(Maybe I'll send pull request after refactor and test it.) https://github.com/syuntoku14/image_based_controls/blob/master/experiments/runner.py

syuntoku14 avatar Feb 04 '20 18:02 syuntoku14