loss的保存

本文探讨了在使用GPU进行深度学习训练时如何避免loss占用过多显存的问题,以及如何将loss数据导出为txt文件。通过将loss的item保存而非直接保存,可以有效防止显存溢出。同时,分享了一个将loss列表转化为txt文件的函数,需要注意的是,由于write函数不接受int类型参数,因此需要先将数据转换为str类型。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、使用gpu时候,loss要保存他的item(),不能直接保存,会极大的占用显存导致报错。
2、导出loss到txt:在trrain函数中写一个list保存,然后打印一下是哪个类型的数据。最后我的是list,并且是float类型,几次报错不可以直接保存。并且需要str转换:
def Save_list(list1,filename):
file2 = open(filename + ‘.txt’, ‘w’)
for i in range(len(list1)):
file2.write(str(list1[i])) # write函数不能写int类型的参数,所以使用str()转化
file2.write(’\n’) # 写完一行立马换行
file2.close()
这个也是根据这个参考的https://blog.youkuaiyun.com/qq_28156907/article/details/103120247

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值