Fix wrong device error created by a breaking change in torch when using DDP - Githubissues

IGNF / myria3d

Myria3D: Aerial Lidar HD Semantic Segmentation with Deep Learning

https://ignf.github.io/myria3d/

BSD 3-Clause "New" or "Revised" License

151 stars 20 forks source link

Fix wrong device error created by a breaking change in torch when using DDP #120

Closed CharlesGaydon closed 2 months ago

CharlesGaydon commented 2 months ago

Cf. https://github.com/IGNF/myria3d/issues/119