Adaptive_avg_pool2d转换失败

[Single430]

你好，非常感谢您的项目！
但是我在代码中有看到Adaptive_avg_pool2d对应的op注册，不过在转换过程报一下错误：

TypeError: adaptive_avg_pool2d() takes 2 positional arguments but 4 were given

我知道是传的参数多了，看代码这里:

caffe_out = raw(input, kernel_size, stride, padding, ceil_mode=True)

应该如何修改呢？
@xxradon @Guanmoyu

[vmc-7645]

Replacing:

caffe_out = raw(input, kernel_size, stride, padding, ceil_mode=True)

With

caffe_out = raw(input, kernel_size)

Worked for me (in removing the error), however the model still didn't seem to work afterwards.