cogment / cogment-verse

Research platform for Human-in-the-loop learning (HILL) & Multi-Agent Reinforcement Learning (MARL)
https://cogment.ai/cogment_verse
Apache License 2.0
78 stars 14 forks source link

Pytorch multiproc fix #81

Closed joshair closed 2 years ago

joshair commented 2 years ago

I tried updating to pytorch 1.12 and it did not fix the problem. It is possible to put the torch.multiprocessing.set_sharing_strategy("file_system") line just once in the torch replay buffer as that is always loaded in. However if the replay buffer is not needed in the future and that import is not longer always there other multiprocesses that use pytorch will likely break.