okankop/Efficient-3DCNNs

depthwise conv not supported

Closed this issue · 2 comments

Hi, thanks for sharing your code. It looks amazing.
But as far as I know, pytorch does not support depthwise convolution yet.

pytorch/pytorch#18631 (comment)

It is actually quite slow compared to standard conv.
Did you figure out a way to make it work?

Depthwise convolutions are not slow only for PyTorch, but also for other platforms like Tensorflow. It is a CUDNN problem since CUDNN is specialized for standard dense convolutions. So I guess we should wait untill someone builds a faster depthwise or group convolutions for CUDNN library.

thank you!