add sampler for multi node - Githubissues

DeepX-inc / machina

Control section: Deep Reinforcement Learning framework

MIT License

279 stars 45 forks source link

add sampler for multi node #235

Closed rarilurelo closed 5 years ago

rarilurelo commented 5 years ago

Using redis to collect episodes

example command

redis-server redis.conf
python -m machina.sampler.distributed_sampler --world_size 2 --rank 0 --redis_host redis_host --redis_port redis_port
python -m machina.sampler.distributed_sampler --world_size 2 --rank 1 --redis_host redis_host --redis_port redis_port
python run_ppo_multi_node_sampler.py --world_size 2

TODO

[x] Add test
[x] Add example
[x] Run in multi node situation (only single node situation is tested)