首页 Paddle框架 帖子详情
pytorch训练的模型,可以使用device_guard接口吗
收藏
快速回复
Paddle框架 问答模型训练 820 0
pytorch训练的模型,可以使用device_guard接口吗
收藏
快速回复
Paddle框架 问答模型训练 820 0

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

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户