search

DingXiaoH / RepVGG

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

Given input size: (64x240x320). Calculated output size: (64x0x1). Output size is too small #77

Open DonMuv opened 2 years ago

DonMuv commented 2 years ago

Hi, There is a error when i use the repvggplus:

File "/home/donmuv/Patch-NetVLAD/repvggplus.py", line 65, in forward out = self.post_se(self.nonlinearity(out)) File "/home/donmuv/anaconda3/envs/NetVlad/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/donmuv/Patch-NetVLAD/se_block.py", line 16, in forward x = F.avg_pool2d(inputs, kernel_size=inputs.size(3)) RuntimeError: Given input size: (64x240x320). Calculated output size: (64x0x1). Output size is too small

YWW-yww commented 2 years ago

Hi,Have you solved this problem? How is it solved?