search

DingXiaoH / RepVGG

RepVGG: Making VGG-style ConvNets Great Again
MIT License
3.31k stars 434 forks source link

jit.trace error #109

Open nistarlwc opened 1 year ago

nistarlwc commented 1 year ago

I want to do "jit.trace " to scripting repvggplus.
But there is a error in "class SEBlock": 

x = F.avg_pool2d(inputs, kernel_size=inputs.size(3))
TypeError: avg_pool2d(): argument 'kernel_size' must be tuple of ints, not Tensor