depthwise conv not supported
Closed this issue · 2 comments
Litou1 commented
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?
okankop commented
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.
Litou1 commented
thank you!