CÃlimos 2023-10-25 14:58 采纳率: 65.5%
浏览 8
已结题

pytorch训练模型时内存不够报错

修改了源码(之前可以正常运行),之前程序读取的标签集合路径是

img

F:\centernet-pytorch-main\VOCdevkit/VOC2007/JPEGImages/00017462fd484921ead12818c60c53b7.jpg 11,115,132,183,0 131,111,693,182,0 ……
这样的格式 其中0表示类别,前面四位是左上和右下的坐标

现在要改成用上四个点的坐标,我重新生成了标签集合文件:

img

F:\centernet-pytorch-main\VOCdevkit/VOC2007/JPEGImages/00017462fd484921ead12818c60c53b7.jpg 15,116,132,115,131,182,11,183,0 132,115,690,111,693,175,131,182,0 从左上右下两点变成了四个点的坐标(为什么改成四个点,因为模型改动里需要用上四个点)
然后把文件里原来读取标签和dataloader的地方都修改了 但感觉还是有问题 :
开始训练的时候在start train卡了好久 第一个epoch都没加载进去 然后报错;Caught MemoryError in DataLoader worker process 0.

看了写网上的教程我把num_workers改成了0 然后就一直卡在这里,直到显示内存不够闪退了

img

感觉是dataloader里加载数据有问题,求高人指点

  • 写回答

2条回答 默认 最新

  • IT工程师_二师兄 2023-10-25 15:01
    关注

    你把代码发给我给你改改

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 1月4日
  • 已采纳回答 12月27日
  • 创建了问题 10月25日

悬赏问题

  • ¥15 需要手写数字信号处理Dsp三个简单题 不用太复杂
  • ¥15 数字信号处理考试111
  • ¥100 关于#audobe audition#的问题,如何解决?
  • ¥15 allegro17.2生成bom表是空白的
  • ¥15 请问一下怎么打通CAN通讯
  • ¥20 如何在 rocky9.4 部署 CDH6.3.2?
  • ¥35 navicat将excel中的数据导入mysql出错
  • ¥15 rt-thread线程切换的问题
  • ¥15 高通uboot 打印ubi init err 22
  • ¥15 R语言中lasso回归报错