syfei0518 2022-04-14 17:11 采纳率: 61.1%
浏览 1202
已结题

yolov5剪枝报错:TypeError: run() got an unexpected keyword argument 'cfg'

问题遇到的现象和发生背景

使用python prune.py --percent 0.5 --weights ./runs/train/exp3/weights/last.pt --data ./data/dataset.yaml --cfg ./models/yolov5s.yaml 命令对yolov5s进行剪枝时报错:TypeError: run() got an unexpected keyword argument 'cfg'

运行结果及报错内容

报错内容如下:

yolo: data=./data/dataset.yaml, weights=['./runs/train/exp3/weights/last.pt'], cfg=./models/yolov5s.yaml, percent=0.5, batch_size=32, imgsz=512, conf_thres=0.001, iou_thres=0.6, task=val, device=, workers=8, single_cls=False, augment=False, verbose=False, save_txt=False, save_hybrid=False, save_conf=False, save_json=False, project=C:\Users\shuyf\yolov5\runs\val, name=exp, exist_ok=False, half=False, dnn=False
test before prune ...
Traceback (most recent call last):
  File "prune.py", line 805, in <module>
    main(opt)
  File "prune.py", line 775, in main
    run(**vars(opt))
  File "C:\ProgramData\Anaconda3\lib\site-packages\torch\autograd\grad_mode.py", line 28, in decorate_context
    return func(*args, **kwargs)
TypeError: run() got an unexpected keyword argument 'cfg'
我的解答思路和尝试过的方法

在网上查询后未发现解决办法

我想要达到的结果

不报错,顺利剪枝

  • 写回答

2条回答 默认 最新

  • 不会长胖的斜杠 后端领域新星创作者 2022-04-14 17:19
    关注

    python prune.py --percent 0.5 --weights ./runs/train/exp3/weights/last.pt --data ./data/dataset.yaml 试试

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

报告相同问题?

问题事件

  • 系统已结题 4月22日
  • 已采纳回答 4月14日
  • 创建了问题 4月14日

悬赏问题

  • ¥15 网络爬虫 在北京新发地抓取数据
  • ¥15 在centos7安装conda
  • ¥15 c#调用yolo3 dll文件获取的数据对不上
  • ¥20 WPF 如何实现多语言,label 和cs(live Charts)中是否都能翻译
  • ¥15 STM32F103上电短路问题
  • ¥15 关于#单片机#的问题:以ATMEGA128或相近型号单片机为控制器设计直流电机调速的闭环控制系统(相关搜索:设计报告|软件设计|流程图)
  • ¥15 打开软件提示错误:failed to get wglChoosePixelFormatARB
  • ¥15 (标签-python|关键词-char)
  • ¥15 python+selenium,在新增时弹出了一个输入框
  • ¥15 苹果验机结果的api接口哪里有??单次调用1毛钱及以下。