Paddle 训练过程中出现内存泄漏

x759pob2  于 2021-12-07  发布在  Java
关注(0)|答案(9)|浏览(759)

 1)PaddlePaddle版本:2.1.0
   2)系统环境:aistudio
在训练前500轮都没有问题,但是在之后的某一时刻就突然停止,出现内存泄漏问题。不清楚是哪方面的问题
在paddle2.0.2版本中没有出现这个问题

rkkpypqq

rkkpypqq1#

您好,我们已经收到了您的问题,会安排技术人员尽快解答您的问题,请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时,您也可以通过查看官网API文档常见问题历史IssueAI社区来寻求解答。祝您生活愉快~

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the APIFAQGithub Issue and AI community to get the answer.Have a nice day!

gg58donl

gg58donl2#

hi,请提供下具体的模型和环境,以便于复现问题。

cbeh67ev

cbeh67ev4#

推测与eval阶段没有使用no_grad有关,我找相关同学check下模型代码写法。

zlhcx6iw

zlhcx6iw5#

好的,谢谢哥哥

j0pj023g

j0pj023g6#

看了下,这个模型是你在AIstudio写的吗?

kxe2p93d

kxe2p93d8#

你可以试下将eval阶段的代码单独拿出来,然后在eval中除了设置model.eval()外,再加上no_grad():

vsaztqbk

vsaztqbk9#

好的,我试一下

相关问题