RuntimeError: Calculated padded input size per channel: (1 x 1). Kernel size: (5 x 5). Kernel size c

使用torchvision里面自带的inception_v3模型时,报了这样的错误。
错误原因是输入图片的大小,此时报错时的图片大小是128 * 128,后来换到256 * 256依然报错,再换到512 * 512就没有报错了。

你可能感兴趣的:(代码,pytorch)