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,两者不匹配所以报错。
#报错
writer.add_image("test_data", imgs,step)
解决办法:用add_images方法替换add_image方法即可。
writer.add_images("test_data", imgs,step)

当遇到AssertionError,提示输入张量尺寸(64,3,32,32)与CHW格式不符时,问题在于数据维度与模型要求的输入格式不一致。解决方案是将`add_image`方法替换为`add_images`,以适应多图像输入的格式,从而正确地在TensorBoard中展示数据。
1961





