search

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

Volume Convolution - Segmentation Fault on Large Sizes #33

Closed av1659 closed 3 years ago

av1659 commented 4 years ago

Hello - For Volume Convolution, in the unit tests, if I change the box_size to 80, batch_size to 2, and number of channels to 48 -- for a volume of (2,48,80,80,80), then I get a segfault in cpu_VolumeConv when I run the Unit Tests. This size is required for some other code (https://github.com/lupoglaz/DeepLocalProteinDocking). Any advice on how to fix this? Thanks!

lupoglaz commented 3 years ago

Moved on with pytorch based fft functions