首页 Paddle框架 帖子详情
飞桨2.0版本调用model.eval之后不再自动关闭反向计算图的构建,怎么解决 已解决
收藏
快速回复
Paddle框架 问答深度学习 437 1
飞桨2.0版本调用model.eval之后不再自动关闭反向计算图的构建,怎么解决 已解决
收藏
快速回复
Paddle框架 问答深度学习 437 1

飞桨2.0版本调用model.eval之后不再自动关闭反向计算图的构建,引入显存的消耗增加,可能会引入OOM,怎么解决?

旧鸱c_
已解决
2# 回复于2021-08
动态图no_grad和 model.eval 解绑,应使用with paddle.no_grad(): 命令,显示关闭反向计算图的构建。
0
收藏
回复
全部评论(1)
时间顺序
旧鸱c_
#2 回复于2021-08

动态图no_grad和 model.eval 解绑,应使用with paddle.no_grad(): 命令,显示关闭反向计算图的构建。

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