pytorch学习
杰酱
这个作者很懒,什么都没留下…
展开
-
PyTorch学习(一)
PyTorch学习(一) 因为实验室整体环境的原因,安装的pytorch版本为0.3.1,而0.4及以上已经做出了比较大的改变,比如: 1.0.4中使用设备:.to(device) 2.0.4中删除了Variable,直接tensor就可以 3.with torch.no_grad():的使用代替volatile 尤其是针对于Variable和tensor的转换,对于新手来说真的是深陷其中费时费力...原创 2019-03-22 21:16:49 · 640 阅读 · 0 评论 -
pytorch:size mimatch,m1:[7168*7],m2:[25088*4096]
目标:迁移vgg16,固定卷积池化层以及前两层全连接层,在分类层的最后一层前面再加两层全连接层 原来的代码: class CNN(nn.Module): def __init__(self, model): super(CNN, self).__init__() self.vgg_layer = nn.Sequential( *list(mo...原创 2019-04-18 21:49:41 · 3545 阅读 · 0 评论