超级会员免费看
在Ubuntu上使用PyTorch进行multi-task learning时，训练至30-60 epoch出现系统卡死现象，分析后确认是由于内存泄漏导致。解决方法是确保在使用SummaryWriter进行数据记录供Tensorboard可视化时，及时关闭SummaryWriter，以防止内存占用过大。
事件：使用pytorch进行multi-task learning，训练到30-60 epoch的时候，机器卡死了。虽然是ubuntu也卡死了
原因：一通没头没脑地分析之后，原因可能是内存泄漏。
解决：
将数据记录到log文件以提供给tensorboard可视化分析的时候，注意要在结束时关闭 SummaryWriter
writer = SummaryWriter(os.path.join(ckptDir, 'logs'))
for epoch in range(num_epochs
订阅专栏 解锁全文
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
