2019年10月20日
[深度学习] Pytorch（三）—— 多/单GPU、CPU，训练保存、加载模型参数问题
摘要： [深度学习] Pytorch（三）—— 多/单GPU、CPU，训练保存、加载预测模型问题 上一篇实践学习中，遇到了在多/单个GPU、GPU与CPU的不同环境下训练保存、加载使用使用模型的问题，如果保存、加载的上述三类环境不同，加载时会出错。就去研究了一下，做了实验，得出以下结论： 多/单GPU训练保 阅读全文
posted @ 2019-10-20 15:04 长颈鹿大侠 阅读(4734) 评论(0) 推荐(0) 编辑
[深度学习] Pytorch学习（二）—— torch.nn 实践：训练分类器（含多GPU训练CPU加载预测的使用方法）
摘要： Learn From: "Pytroch 官方Tutorials" "Pytorch 官方文档" 环境：python3.6 CUDA10 pytorch1.3 vscode+jupyter扩展 结果： 阅读全文
posted @ 2019-10-20 09:22 长颈鹿大侠 阅读(1217) 评论(0) 推荐(0) 编辑