search

octree-nn / ocnn-pytorch

Octree-based 3D Convolutional Neural Networks
MIT License
156 stars 17 forks source link

Fix AMP dtype autocast #40

Closed Gofinge closed 2 months ago

wang-ps commented 2 months ago

Thanks for fixing this issue! @Gofinge