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

Updated main.cu #22

Closed pulkit119 closed 5 years ago

pulkit119 commented 5 years ago

A raw attempt at using 16*16 (+2 stencil for computation) Slices in x,y plane first and then evaluated in 3D