2020年6月1日
[python][pytorch]多GPU下的模型保存与加载
摘要： 说明 在模型训练的时候，往往使用的是多GPU的环境；但是在模型验证或者推理阶段，往往使用单GPU甚至CPU进行运算。那么中间有个保存和加载的过程。下面来总结一下。 多GPU进行训练 首先设置可见的GPU数量，有两种方式可以声明： 在shell脚本中声明： export CUDA_VISIBLE_DE 阅读全文
posted @ 2020-06-01 15:13 wildkid1024 阅读(4630) 评论(0) 推荐(1) 编辑