mindspore.JitConfig

class mindspore.JitConfig(jit_level='O1', exc_mode='auto', **kwargs)

编译时所使用的JitConfig配置项。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• jit_level (str) - 设置编译优化的级别，支持[“O0”, “O1”, “O2”, “O3”]。默认值： "O1" 。

  • “O0”: 基础优化。

  • “O1”: 手动优化。

  • “O2”: 手动优化与图算优化结合。

  • “O3”: 性能优化，无法保证泛化性。

• exc_mode (str) - 设置执行模式，支持[“auto”, “sink”, “no_sink”]。默认值： "auto" 。

  • “auto”: 自动策略。

  • “sink”: 计算图下沉策略。

  • “no_sink”: 非计算图下沉策略。

• kwargs (dict) - 关键字参数字典。

样例：

>>> from mindspore import JitConfig
>>>
>>> jitconfig = JitConfig(jit_level="O1")
>>>
>>> # Define the network structure of LeNet5. Refer to
>>> # https://gitee.com/mindspore/docs/blob/r2.1/docs/mindspore/code/lenet.py
>>> net = LeNet5()
>>>
>>> net.set_jit_config(jitconfig)