pytorch训练的模型,可以使用device_guard接口吗
收藏
pytorch训练的模型,可以使用device_guard接口吗?PyTorch 采用的是动态图机制,device_guard的api仅支持静态图模式,需要把动态图先转为静态图吗?“在模型训练或者推理部署时,只需添加一行装饰器 @to_static,即可将动态图代码转写为静态图代码,并在底层自动使用静态图执行器运行。 可获得更好的模型运行性能。”“动静转换(@to_static)通过解析 Python 代码(抽象语法树,下简称:AST) 实现一行代码即可将动态图转为静态图的功能,只需在待转化的函数前添加一个装饰器 @paddle.jit.to_static 。”使用这条代码@paddle.jit.to_static 就可以转换吗?
0
收藏
请登录后评论