TensorBoard图片显示不完整问题咨询（附PyTorch代码）

问题原因 #

1. TensorBoard默认网格布局自动调整：使用add_images传入64张图片时，TensorBoard默认会自动排列为8×8的网格，但如果界面缩放比例不当，或是CIFAR10的32×32小尺寸图片，容易出现显示截断、挤压的视觉异常。
2. 最后一个batch样本数不足：CIFAR10测试集共10000张样本，batch_size=64且drop_last=False时，最后一个batch仅包含16张图片，对应TensorBoard中该step会显示4×4的网格，易被误认为显示异常。
3. TensorBoard版本兼容性：旧版TensorBoard可能存在图片渲染bug，导致显示不全。

解决方案 #

• 手动调整TensorBoard网格布局：在TensorBoard的图片查看界面，点击右上角的设置图标，手动指定网格的行数和列数（比如设置为8行8列），即可让64张图片完整显示。
• 保证每个batch样本数一致：将DataLoader的drop_last参数设为True，自动丢弃最后一个不足64张的batch，所有step显示的图片均为64张。修改后的代码片段：

  test_loader = DataLoader(test_data, batch_size=64, shuffle=False, num_workers=0, drop_last=True)
  

• 升级TensorBoard到最新版本：执行以下命令修复旧版可能存在的渲染bug：

  pip install --upgrade tensorboard
  

额外验证 #

你的代码中add_images的张量维度完全正确（(batch_size, 3, 32, 32)符合NCHW格式），无需调整数据维度；第一个循环遍历单样本的add_image逻辑也正常，仅需针对批量图片的显示问题调整上述项即可。

内容的提问来源于stack exchange，提问作者锦丝虾球