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)