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模型内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复