多卡下专题

PyTorch单卡/多卡下模型保存/加载

由于训练和测试所使用的硬件条件不同,在模型的保存和加载过程中可能因为单GPU和多GPU环境的不同带来模型不匹配等问题。这里对PyTorch框架下单卡/多卡下模型的保存和加载问题进行排列组合(=4),样例模型是torchvision中预训练模型resnet152,不尽之处欢迎大家补充。   1 数据格式与存储内容 1.1 模型存储格式 PyTorch存储模型主要采用pkl,pt,pth三种格