2019年10月21日
PyTorch 多GPU下模型的保存与加载（踩坑笔记）
摘要： 这几天在一机多卡的环境下，用 训练模型，遇到很多问题。现总结一个实用的做实验方式： 多GPU下训练，创建模型代码通常如下： 官方建议的模型保存方式，只保存参数： 其实，这样很麻烦，我建议直接保存模型（参数+图）： 这样做很实用，特别是我们需要反复建模和调试的时候。这种情况下模型的加载很方便，因为模型 阅读全文
posted @ 2019-10-21 00:26 叶罅 阅读(20387) 评论(0) 推荐(0) 编辑