概述
import torch
def convert_model(para_model,single_model):
checkpoint = torch.load(para_model ,map_location=torch.device("cpu"))
output={}
for key,value in checkpoint['model'].items():
output[key.lstrip("module.")]=value
torch.save({'model':output},single_model)
if __name__=="__main__":
convert_model(para_model="model_2.pth", single_model="model.pth")
最后
以上就是怕黑乌冬面为你收集整理的将gpu并行模型转换为单cpu/gpu模型的全部内容，希望文章能够帮你解决将gpu并行模型转换为单cpu/gpu模型所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复