Do this work with the 'deformable convolution' as well?

[dheerajvarma24]

If our model has a deformable convolution conv then can we still use this library?

[sovrasov]

That depends on implementation you're using

[Koubunnkei]

I'd like to know the answer to that question as well. May I ask which implementation guarantees that complexity can be calculated using this library?

[dheerajvarma24]

What worked for me is deepspeed's flops-profiler

[sovrasov]

I'd like to know the answer to that question as well. May I ask which implementation guarantees that complexity can be calculated using this library?

DeformConv2d from torchvision is supported

[pvti]

What worked for me is deepspeed's flops-profiler

Thank you very much for your comment. I'm seeking a package that covers the torch.nn.functional.conv2d and found your comment. I tried deepspeed and it works like a charm.