Encoder-Decoder

用nn.Sequential()编写神经网络结构：https://blog.csdn.net/Lzj000lzj/article/details/94997448

用Pytorch加载指定文件夹的图文件：https://blog.csdn.net/weixin_41514525/article/details/109300091

ERROR1：

ValueError错误：

• 错误原因：
  • 上述错误由nn.BatchNorm2d层抛出，产生的原因涉及两个层：
  • class Splat(nn.Module)中全局池化层，即代码gap = F.adaptive_avg_pool2d(gap, 1)；
  • nn.BatchNorm2d层：该层在training时需要input shape:[batch, c, h, w]满足：batch * h * w > 1；<该条件是从torch.nn.functional.batch_norm函数中推导出来的>
  • 而经过全局池化层后，h=w=1，若此时batch=1，则不满足batch * h * w > 1,因此会抛出错误。

• 解决错误：设置batch > 1，且 drop_last = True。

作者：zhangpanPP
链接：https://www.jianshu.com/p/2b9c0279e1d4
来源：简书

ERROR2：

images, _ = next(iter(loader))  # 此处要求loader中的batch_size 基于等于所给数据集的数量

ERROR3:

tb.add_graph(network, images) # 但用此句可能显示不出grapth

更正：

with tb:
    tb.add_graph(network, images)