yolov5中train.py运行配置以及报错解决方案

1.依赖库版本要求

yolov5中不同版本的依赖库要求可以在requirements.txt查看。根据你的yolov5版本,选择好满足条件的版本。最新版本链接:

yolov5/requirements.txt at master · ultralytics/yolov5 (github.com)

2.parser = argparse.ArgumentParser()参数配置

这里只需要更改--data的参数,改成训练数据的路径,如下图:

当然也可以在代码中直接修改:

 3.报错信息:FailedPreconditionError: runs is not a directory

解决方案: 删除runs文件夹

4.报错信息: OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading "C:\Users\NL\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\lib\cudnn_adv_train64_8.dll" or one of its dependencies.

原因:batch_size太大
大概在410行处

解决方法,将batch_size改成8或者更小 

5.报错信息:Tensor******

因为有些数据不是tensor数据,这个要慢慢看调试改错

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值