深度学习
实用干货:如何把Pytorch模型参数加载 …
问题简述 在日常的模型开发、训练过程中我们经常会遇到这样的现象:在现有的开源项目或者论文复现中,多数模型使用Pytorch设计、开发和训练推理,当我们需要使用MindSpore框架进行模型开发时,会遇到以下两个问题: 模型使用Pytorch编码; Pytorch模型训练后保存的参数无法被MindSpore模型直接加载。 对于第一个问题,我们可以根据昇思官方提供的文档:与Pytorch典型区别和PyTorch与MindSpore API映射表来完成模型的迁移; 对于模型参数的转换,在最新的MindSpore版本中MindConverter不再支持,因此可以考虑针对模型参数,我们进行手动的转换, …