分布式训练模型保存torch.save()、加载model.load_state_dict常见问题:Missing key(s) in state_dict,Unexpected key()解析

一块GPU训练的模型:

保存

torch.save(model.state_dict(), PATH)

保存的keys格式如下:
在这里插入图片描述

加载到一块GPU/CPU的方法:

device = torch.device('cpu')
model = TheModelClass(<

你可能感兴趣的:(Machine,&,Deep,Learning,深度学习,人工智能)