pytorch使用过程碰到的错误
世纪末的霖さん
这个作者很懒,什么都没留下…
展开
-
pytorch出现UserWarning: torch.range is deprecated in favor of torch.arange and will be removed in 0.5.
今天使用torch出现了这样的警告, UserWarning: torch.range is deprecated in favor of torch.arange and will be removed in 0.5. Note that arange generates values in [start; end), not [start; end]. 代码如下: a = torch.ran...原创 2019-08-28 10:54:16 · 3173 阅读 · 0 评论 -
自学pytorch深度学习遇到的坑之模型加载出错
当保存了神经网络模型后,进行训练加载模型的时候一直出错,代码如下: model = model.load_state_dict(torch.load(path)) 运行结果如下,出现: Error(s) in loading state_dict for ResNet18 出现这样的问题,应该是版本的原因,这样的语句现在新的版本好像不支持了(如果有朋友知道具体的原因,欢迎留言指正哈……) 这...原创 2019-09-09 09:10:02 · 2460 阅读 · 0 评论 -
谷歌云盘colab深度学习遇到的问题
在代码里面写文件路径的时候,应该这样写: /content/drive/“你自己的文件夹的名字” 之前没有这样写报错 :没有这个路径 谷歌这个免费的平台用起来确实略麻烦,不过谁叫自己买不起显卡呢!后面用的时候又发现了一个问题,就是import 其他自己写的文件模块,提示没有这个模块,折腾了两天,应该是解决了,但是后面我还是把几个文件写成一个文件,虽然长一点,但是不用每个文件都要挂载一遍...原创 2019-09-12 08:54:20 · 805 阅读 · 4 评论 -
pytorch迁移学习自己的理解
在学习深度学习的过程中,迁移学习到现在还是理解的不太好,为此,把想到的写下来,无论正确与否,和大家有个交流,促进学习。 1. trained_model=resnet18(pretrained=True) # 此时是一个非常好的model model = nn.Sequential(*list(trained_model.children())[:-1],Flatten(),nn.Linear(...原创 2019-09-12 11:22:53 · 236 阅读 · 0 评论