Skip to content

feixiao15/torch-ms-lab

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch vs MindSpore 模型一致性实验总结

本实验对 AlexNet、GoogLeNet、ResNet、MobileNetV2 在 PyTorch 与 MindSpore 下的实现进行对比,主要通过相同输入下 logits 误差论证两框架的模型一致性:当 logits 最大/平均误差在 1e-2~1e-3 量级、预测一致率接近或达到 100% 时,可认为权重转换与前向推理等效。


1. AlexNet

指标 数值
数据集 ImageNet2012val
logits 最大误差 1.40e-02
logits 平均误差 1.24e-03
预测一致率 100.0%
PyTorch 准确度 56.40%
MindSpore 准确度 56.40%

2. GoogLeNet

指标 数值
数据集 ImageNet2012val
logits 最大误差 3.48e-02
logits 平均误差 2.51e-03
预测一致率 99.2%
PyTorch 准确度 60.80%
MindSpore 准确度 60.80%

3. MobileNetV2

指标 数值
数据集 ImageNet2012val
logits 最大误差 2.57e-02
logits 平均误差 3.17e-03
预测一致率 100.0%
PyTorch 准确度 69.20%
MindSpore 准确度 69.20%

4. ResNet

指标 数值
数据集 CIFAR-10 测试集
logits 最大误差 2.42e-02
logits 平均误差 3.56e-03
PyTorch 准确率 91.09%
MindSpore 准确率 91.09%

结论

  • logits 误差:四模型在两框架下的 logits 最大误差均在 1e-2~3e-2 量级、平均误差在 1e-3 量级,属浮点运算正常差异。
  • 预测一致:AlexNet、MobileNetV2 预测一致率 100%,GoogLeNet 99.2%,Top-1 预测高度一致。
  • 准确度一致:各模型 PT/MS Top-1 准确度完全相同,验证权重转换与推理逻辑正确。

综合上述结果,可认为 PyTorch 与 MindSpore 的模型实现及权重转换具有良好一致性。

About

Convert PyTorch vision models to MindSpore and verify parity via logits comparison.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages