YOLOv5训练模型求助帖

有没有大神帮我看一下,这个错误到底怎么修改啊。在用yolov5训练模型的时候,一直报错,然后修改了dataloaders.py,loss.py等,会不会越改越乱啊,我看网上都没有说训练的时候要修改这些文件呜呜呜。

目前的一个错误是这样的:

Traceback (most recent call last):
  File "D:\yolov5-7.0\train.py", line 634, in <module>
    main(opt)
  File "D:\yolov5-7.0\train.py", line 528, in main
    train(opt.hyp, opt, device, callbacks)
  File "D:\yolov5-7.0\train.py", line 351, in train
    results, maps, _ = validate.run(data_dict,
  File "D:\anaconda\envs\yolov5\lib\site-packages\torch\autograd\grad_mode.py", line 28, in decorate_context
    return func(*args, **kwargs)
  File "D:\yolov5-7.0\val.py", line 213, in run
    loss_part = compute_loss(train_out, targets)
  File "D:\yolov5-7.0\utils\loss.py", line 106, in __call__
    obji = self.BCEobj(pi[..., 4], tobj_clone)
  File "D:\anaconda\envs\yolov5\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
    return forward_call(*input, **kwargs)
  File "D:\anaconda\envs\yolov5\lib\site-packages\torch\nn\modules\loss.py", line 713, in forward
    return F.binary_cross_entropy_with_logits(input, target,
  File "D:\anaconda\envs\yolov5\lib\site-packages\torch\nn\functional.py", line 2960, in binary_cross_entropy_with_logits
    return torch.binary_cross_entropy_with_logits(input, target, weight, pos_weight, reduction_enum)
RuntimeError: Inplace update to inference tensor outside InferenceMode is not allowed.You can make a clone to get a normal tensor before doing inplace update.See https://github.com/pytorch/rfcs/pull/17 for more details.

进程已结束,退出代码1
 

非常感谢大家百忙之中给我看看,能不能给我个修改建议啊!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值