RuntimeError: Given input size: (2048x1x1). Calculated output size: (2048x-5x-5). Output size is too small
bitwangdan opened this issue · 1 comments
bitwangdan commented
Hi， when i run "python main.py --cfg ./config/CIFAR10_LT/causal_norm.yaml", i got this error:
RuntimeError: Given input size: (2048x1x1). Calculated output size: (2048x-5x-5). Output size is too small
but when i run "python main.py --cfg ./config/CIFAR10_LT/causal_norm_32.yaml", it works
Have you ever encountered this problem?
chastloveRYU commented
change the kernel size of averagepool