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