热门
最新
红包
立Flag
投票
同城
我的
发布
pytorch中保存网络参数,利用训练好的网络推理,利用已知网络参数初始化下一次训练的网络,迁移学习等等。可以参考这一篇
https://zhuanlan.zhihu.com/p/417997662
需要注意的是,在训练完网络以后,利用网络去做推理时,应设置model.eval()和with torch.no_grad()
https://www.cnblogs.com/BlueBlueSea/p/14616416.html
目的是让BN层和dropout层失效,不然的话,会造成严重后果
https://blog.csdn.net/wuqingshan2010/article/details/106013660
CSDN App 扫码分享
评论
点赞
打赏
- 复制链接
- 举报