深度学习:size of input tensor and input format are different

错误代码:

 writer.add_image("input", imgs, step,  dataformats='HWC')

正确代码:

 writer.add_images("input", imgs, step,  dataformats='HWC')

错误原因:

add_image只接收单一图像,而我们直接将imgs批量传入是不行的。

你可能感兴趣的:(实战深度学习常见问题汇总,python,人工智能,深度学习,迁移学习,pytorch)