深度学习踩坑记录

深度学习踩坑记录

在跑一个深度学习的项目的时候,为了计算每个epoch的Loss,肯定要把每个batch_size的loss给加起来,就会有类似如下的代码

train_loss += loss

是的当时手抖,忘了是loss.item(),代码也能正常运行,也能正常算,并且每一步需要跑的时间也是一样的,但是这样会导致每一步的内存都是叠加的。并且这种占显存的方式你在nvidia-smi还看不出来,得去htop里面看,一看好家伙,显存占的高的离谱,正常来说我就占4个G,他占了十几个G,当时就感觉不对劲了,后面发现这叫显存泄露
总而言之言而总之,如果你直接+loss,也能跑,结果也是一样的,但是这就会导致需要的内存一步步的叠加,我是跑到10几个epoch的时候莫名其妙被killed了,去查了一下是out of memory的问题,刚开始还以为是别的因素,后面才发现是我的问题T T

相关推荐

  1. 深度学习记录

    2023-12-13 18:06:01       38 阅读
  2. 深度学习记录(持续更新)

    2023-12-13 18:06:01       18 阅读
  3. mySQL记录

    2023-12-13 18:06:01       43 阅读
  4. Pinia 记录

    2023-12-13 18:06:01       33 阅读
  5. golang记录

    2023-12-13 18:06:01       18 阅读
  6. NLP学习记录(持续更新版)

    2023-12-13 18:06:01       7 阅读

最近更新

  1. TCP协议是安全的吗?

    2023-12-13 18:06:01       16 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2023-12-13 18:06:01       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2023-12-13 18:06:01       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2023-12-13 18:06:01       18 阅读

热门阅读

  1. 用五分钟了解网络安全404页面检测

    2023-12-13 18:06:01       35 阅读
  2. Go 语言函数

    2023-12-13 18:06:01       50 阅读
  3. (第1天)保姆级 VirtualBox 虚拟机安装 Linux 系统

    2023-12-13 18:06:01       40 阅读
  4. C_8练习题答案

    2023-12-13 18:06:01       31 阅读
  5. C#面试基础语法&和&&的区别

    2023-12-13 18:06:01       34 阅读
  6. 每日一题,狒狒吃香蕉

    2023-12-13 18:06:01       39 阅读