model = torchvision.models.resnet18(pretrained=True)

飘香一剑 · 发表于 2019-3-8 12:01:25

您需要登录才可以下载或查看，没有账号？立即注册

x

这个代码运行时后下载数据，但是网速很慢。我已经在别的地方下载了数据放在了当前文件夹里面。但运行程序时还是会下载，怎么办？

前路 · 发表于 2019-3-8 13:33:19

没有太明白你的问题描述！能具体一点吗？

Charles未晞 · 发表于 2019-3-8 15:18:32

model = torchvision.models.resnet18(pretrained=False)
model.load_state_dict(torch.load(权重路径))

这样写

飘香一剑 · 发表于 2019-3-8 23:07:29

这样不是照样下载吗？我已经下载好了，放在了当前文件夹里面，就是要修改掉这个下载的代码。我在你的微信公众号里面问过你的，

飘香一剑 · 发表于 2019-3-9 22:06:48

Charles未晞发表于 2019-3-8 15:18
这样写

会不会是因为我没有设置GPU训练环境

账号		自动登录	找回密码
密码			立即注册