Closed rarilurelo closed 5 years ago
TODO:
[x] Incorporate ddp (distributed data parallel) to existing models (models, pols, vfuncs)
[x] Add synchronizing methods to Traj
[x] DDP with cloudpickle is incompatible. Making another solution
[x] Change sampling scheme of Traj's iterator something like DistributedSampler
[x] Add example
[x] Add tests
[x] Test in multi node multi gpu situation
TODO:
[x] Incorporate ddp (distributed data parallel) to existing models (models, pols, vfuncs)
[x] Add synchronizing methods to Traj
[x] DDP with cloudpickle is incompatible. Making another solution
[x] Change sampling scheme of Traj's iterator something like DistributedSampler
[x] Add example
[x] Add tests
[x] Test in multi node multi gpu situation