AssertionError: size of input tensor and input format are different.

转载:本文链接:https://blog.csdn.net/lin958490203/article/details/118995944

在写代码的时候出现以下错误:AssertionError: size of input tensor and input format are different. tensor shape: (64, 3, 32, 32), input_format: CHW

#原因:tensor的维度是(64,3,32,32)而要求的输入格式是CHW,两者不匹配所以报错。

解决办法:用add_images方法替换add_image方法即可。

你可能感兴趣的:(Error_pytorch,python)