【Mindspore模型迁移指北】

文章目录

  • 1. Mindspore算子
    • 1.1 Mindspore.nn算子
    • 1.2 Mindspore.ops算子
    • 1.3 Mindspore隐式类型转换算子
    • 1.4 Mindspore分布式算子
  • 2. Mindspore与Tensorflow/Pytorch的API映射关系
  • 3. Mindspore网络支持
  • 4. Mindspore网络迁移指南

【笔者按】 笔者最近在进行讲Mask_RCNN网络迁移到Mindspore框架的工作。本文会按照个人进度迭代更新,记录所学所思。

1. Mindspore算子

Mindspore支持的算子

1.1 Mindspore.nn算子

mindspore.nn
神经网络Cell。用于构建神经网络中的预定义构建块或计算单元。

1.2 Mindspore.ops算子

mindspore.ops
可用于Cell的构造函数的算子。

import mindspore.ops as ops

1.3 Mindspore隐式类型转换算子

mindspore隐式类型转换算子
【转换规则】

  • 标量与Tensor运算:运算时,将标量自动转为Tensor,数据类型和参与运算的Tensor数据类型保持一致;当Tensor是bool数据类型,标量是int或float时,将标量和Tensor都转为数据类型为int32或float32的Tensor;当Tensor是int或者uint数据类型,标量是float时,将标量和Tensor都转为数据类型为float32的Tensor。
  • 不同数据类型Tensor运算:数据类型优先级排序为bool < uint8 < int8 < int16 < int32 < int64 < float16 < float32
  • 不支持对Parameter进行数据类型转换:如果按照转换规则推导,需要对网络中定义的Parameter进行数据类型转换时,会抛出RuntimeError异常。

1.4 Mindspore分布式算子

mindspore分布式算子

2. Mindspore与Tensorflow/Pytorch的API映射关系

Mindspore与Tensorflow的API映射
Mindspore与Pytorch的API映射

import tensorflow as tf
import mindspore as ms

tf.cast(scalar,tensorflow.dtype)<--->ms.ops.ScalarCast(scalar,mindspore.dtype)

3. Mindspore网络支持

Mindspore支持的网络目录

4. Mindspore网络迁移指南

Mindspore迁移指南

【Mindspore模型迁移指北】_第1张图片

你可能感兴趣的:(深度学习,python,神经网络,华为,tensorflow)