pytorch 解决训练完成后读取模型进行测试时,执行enumerate函数后模型权重突然全部清零的问题

在进行测试时读取模型后,发现权重有正常读取进来:

然后进行执行,当执行到enumerate语句后:

Pycharm会卡顿一顿时间,然后会发现模型的权重全部清零了,导致测试结果全部一致为某个数。

调试后发现执行到dataloader.py文件里的w.start()时,pycharm会卡顿:

然后发现这里是与多线程有关的,推测可能是多线程的原因,然后试着修改配置文件里的num_workers的数值:

原本是3,将它修改为0:

问题得到解决,pycharm也不会突然卡顿,模型参数也不会清零了。

  • 4
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值