接连下了两天的细雨,伴随着阵阵落雨声,今天学习了第十课——使用静态图加速。
AI编译框架分为两种运形模式,分别是动态图模式以及静态图模式。MindSpore默认情况下是以动态图模式运行但也支持手工切换为静态图模式。
动态图模式的特点,是计算图的构建和计算同时发生,符合python的解释执行方程。
静态图的特点是将计算图的构建和实际计算分开,有关静态图模式的运行原理可以参考静态图语法支持。
静态图模式的开启方式,分别是基于装饰器的开启方式和基于全集context的开启方式。
基于装饰器的开启方式,在需要对Tensor的某些运算进行编译加速时,可以在其定义的函数上使用jit修饰器。在调用该函数时,该模型自动被编译为静态图。需要注意的是,此装饰器只能用来修饰函数,无法对类进行修饰。
基于context 的开启方式,context是一种全局的设置模式 。