search

1zb / deformable-convolution-pytorch

PyTorch implementation of Deformable Convolution
409 stars 53 forks source link

Slow speed #14

Open zhujiagang opened 6 years ago

zhujiagang commented 6 years ago

An excellent you have done! But I don't know why it runs very slow when it is embeded into my network. I also refer to the implementation of https://github.com/ChunhuanLin/deform_conv_pytorch, pure pytorch without c, cu files, which is faster than yours but costs more memory.

aachenhang commented 5 years ago

Hi,zhujiagang, Could you please share how much time you have cost to train 1zb's DCN and ChunhuanLin's DCN? Currently I'm training my network with ChunhuanLin's DCN, however I found it cost to mush memory cause it copy the previous feature map of the DCN by 9 times, so that it can add a normal conv with stride=kernel_size to implement DCN.

aachenhang commented 5 years ago

Sorry for my unclearly words, I mean how much time it takes in one batch in 1zb's DCN and ChunhuanLin's DCN?