lamoureux-lab / TorchProteinLibrary

PyTorch library of layers acting on protein representations
https://lamoureux-lab.github.io/TorchProteinLibrary/
MIT License
116 stars 23 forks source link

RMSD error about function not implemented #36

Closed yuanqidu closed 3 years ago

yuanqidu commented 3 years ago

FullAtomModel/CoordsTransform/CoordsTransform.py", line 174, in forward _FullAtomModel.Coords2CenterGPU_forward( input_coords, output_center, num_atoms) RuntimeError: "Coords2CenterGPU_forward" not implemented for 'Half'

I am confused by why this happened. It looked like the function is not implemented?

yuanqidu commented 3 years ago

NVM, it was due to mixed-precision training.

lupoglaz commented 3 years ago

Yeah, right now there are no layers that are using half precision.