![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
zpkosmos
这个作者很懒,什么都没留下…
展开
-
EDSR的pytorch代码修改后的plot报错问题
应该是epoch的计数方式不对造成的 train的代码里应该是 epoch = self.optimizer.get_last_epoch() + 1 test的代码里应该是 epoch = self.optimizer.get_last_epoch() 有时候+1的位置会错原创 2020-07-06 22:35:05 · 437 阅读 · 2 评论 -
pytorch的一些坑01:用预训练的vgg模型的部分层的特征报错,如张量不匹配
#打算取VGG19的第二个全连接层的输出,那么就需要构建一个类,这个类要包含VGG的全部卷积层, #以及到第二个全连接层的全部网络还有他们对应的参数 class Classification_att(nn.Module): def __init__(self, rgb_range): super(Classification_att, self).__init__() ...原创 2019-11-04 15:52:42 · 2722 阅读 · 0 评论