在使用tensorboard时报错,AssertionError: size of input tensor and input format are different. tensor shape: (1, 3, 224, 224), input_format: CHW
错误通常表示您调用了一个期望输入数据格式与实际数据格式不匹配的函数,input_format: CHW
是期望的输入格式,它表示张量的维度顺序为通道(C)、高度(H)和宽度(W)。
这里img来自dataloader,这个dataloader设置的batchsize为1,即一批次一张图片,因此刚开始使用add_image()函数加载图片。
报错后仔细看img.shape发现,即使batchsize为1,图片格式也不是CHW,还是要使用add_images()加载图片
![](https://i-blog.csdnimg.cn/blog_migrate/3ab96b37dbdb53d712c5b0b17e539a54.png)