天天看点

MindSpore保存模型的格式疑惑

请问MindSpore保存模型的格式和TensorFlow一样为.ckpt吗?

那导出为onnx格式后进行模型转换的.om格式又是什么?

解答:

MindSpore支持保存两种类型的数据:训练参数和网络模型(模型中包含参数信息)。

  • 训练参数指的是Checkpoint格式文件。
  • 网络模型包括MindIR、AIR和ONNX三种格式文件。

更多详细信息请参考:

https://www.mindspore.cn/tutorials/experts/zh-CN/master/infer/inference.html

继续阅读