Runtime error when running uniform_test.py

[yrajas]

The same error happens when running with resnet18 as well. Appreciate any suggestions to get this fixed.

$ python ./uniform_test.py --dataset cifar10 --model mobilenetv2_w1 ****** Full precision model loaded ****** Files already downloaded and verified Traceback (most recent call last): File "./uniform_test.py", line 75, in dataloader = getDistilData( File "/home/user/ZeroQ/classification/distill_data.py", line 121, in getDistilData output = teacher_model(gaussian_data) File "/home/user/env3/env/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/home/user/env3/env/lib/python3.8/site-packages/pytorchcv/models/mobilenetv2.py", line 147, in forward x = self.features(x) File "/home/user/env3/env/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/home/user/env3/env/lib/python3.8/site-packages/torch/nn/modules/container.py", line 141, in forward input = module(input) File "/home/user/env3/env/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/home/user/env3/env/lib/python3.8/site-packages/torch/nn/modules/pooling.py", line 616, in forward return F.avg_pool2d(input, self.kernel_size, self.stride, RuntimeError: Given input size: (1280x1x1). Calculated output size: (1280x-5x-5). Output size is too small