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
loss的保存
最新推荐文章于 2024-04-28 23:51:00 发布