昇思大模型平台学习第九天

MindSpore 支持静态图 和动态图切换 默认是动态图

1.全局切换

# 使用set_context进行动态图模式的配置
ms.set_context(mode=ms.PYNATIVE_MODE) 

# 使用set_context进行静态图模式的配置
ms.set_context(mode=ms.GRAPH_MODE) 

2.1 使用装饰器方式切换静态图  装饰器只能在函数上调用 无法在类调用

@ms.jit  # 使用ms.jit装饰器，使被装饰的函数以静态图模式运行
def run(x):
    model = Network()
    return model(x)

2.2 使用函数方式切换静态图

run_with_jit = ms.jit(run)  # 通过调用jit将函数转换为以静态图方式执行
output = run_with_jit(input)