facebookresearch / torchbeast

A PyTorch Platform for Distributed RL
Apache License 2.0
738 stars 114 forks source link

flags.num_servers are fixed to default #15

Open urw7rs opened 4 years ago

urw7rs commented 4 years ago

flags.num_servers in polybeast_env should be set to flags.num_actors in polybeast.py before passing it to polybeast_env.main().

Adding flags.num_servers = flags.num_actors before line 45 in polybeast.py fixes hang on actorpool_thread.join() when terminating with Ctrl-C.

I would have added a pull request but I already modified the forked repo so I'm not sure how to add a pull request.