请问训练完了如何释放显存
收藏
框架: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
收藏
请登录后评论
手动的话建议重启执行器
点项目里那个重启按钮
建议不要用AI Studio