技术栈
模型的保存和加载
小白冲鸭
4 个月前
人工智能
·
pytorch
·
python
·
gpu
·
模型的保存和加载
pytorch使用DataParallel并行化保存和加载模型(单卡、多卡各种情况讲解)
话不多说,直接进入正题。!!!不过要注意一点,本文保存模型采用的都是只保存模型参数的情况,而不是保存整个模型的情况。一定要看清楚再用啊!