jit.trace error

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
```