首页 Paddle框架 帖子详情
请问训练完了如何释放显存
收藏
快速回复
Paddle框架 问答模型训练炼丹技巧 1175 3
请问训练完了如何释放显存
收藏
快速回复
Paddle框架 问答模型训练炼丹技巧 1175 3

框架:paddle2.0.2

问题: 

       不同权重的组合损失(eg: Loss = loss1+alpha*loss2; alpha = [0.1, 0.2, 0.3])想一次性把三个都跑完。

      但是,每次结束一轮,到第二轮的时候就会GPU out of Memory!

      请教一下各位小伙伴,怎么手动释放显存呀。。。

#
for a in alpha:
   # a[1]=0.1; a[2]=0.2; a[3]=0.3
   net = Model();
   
   for epoch in range(10):
       loss = net(input)
       loss.backward()
      
   #完成一轮,保存参数
   net.save_dict()

0
收藏
回复
全部评论(3)
时间顺序
三岁
#2 回复于2021-08

手动的话建议重启执行器

0
回复
FutureSI
#3 回复于2021-08

点项目里那个重启按钮

0
回复
Fish fish
#4 回复于2023-07

建议不要用AI Studio

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