首页 Paddle框架 帖子详情
显存爆炸的问题 a[a>=b]=1
收藏
快速回复
Paddle框架 其他炼丹技巧深度学习 1078 4
显存爆炸的问题 a[a>=b]=1
收藏
快速回复
Paddle框架 其他炼丹技巧深度学习 1078 4

a[a>=b]=x  会导致显存一直积累, 可能是一直产生a 的拷贝

使用 paddle.where(a>=b, x1, x2) 来替代

0
收藏
回复
全部评论(4)
时间顺序
耀
耀光17
#4 回复于2021-11

厉害

0
回复
耀
耀光17
#5 回复于2021-11
厉害

厉害

0
回复
耀
耀光17
#6 回复于2021-11
厉害

厉害

0
回复
耀
耀光17
#7 回复于2021-11

厉害

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