code出错
一位以泪洗面的同学
这个作者很懒,什么都没留下…
展开
-
ValueError: Expected more than 1 value per channel when training, got input size[1,x,1,1]
今天使用pytorch对模型加载,出现了ValueError: Expected more than 1 value per channel when training, got input size[1,x,1,1]问题。按字面理解就是训练时,期望每个通道上超过1个值。当我们在dataloader时,batch_size划分到最后,余数为1时, 我们输入大小在变换过程中成为了(1, num_channel, 1, 1)。 此时,如果计算batch Norm时, 它仅含有1个值。 即图中的一个小立方原创 2021-04-14 19:00:52 · 413 阅读 · 0 评论 -
使用pytorch进行模型加载时,遇见IncompatibleKeys‘ object has no attribute ‘xxx‘问题
今天使用pytroch进行模型加载,出现了IncompatibleKeys' object has no attribute 'xxx'问题,该问题发生情况如下。错误案例:def fun() model = VGG16() if pretrained: model = model.load_state_dict(torch.load('xxx.pth')) return model会报错_IncompatibleKeys' object has no attribu原创 2020-07-26 15:55:58 · 8866 阅读 · 2 评论