×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

yolov5_v6量化出现问题 已完结 gxkj2023-07-27 15:49:05 回复 3 查看 技术答疑 使用求助
yolov5_v6量化出现问题
分享到:

使用官方文档进行操作,在docker环境下进行量化 python mlu_forward.py文件,出现报错forward参数问题,但是检查报错中的commn.py和yolo.py文件里的forward参数无问题,严格按照文档操作的,pt文件的md5也能对上,求助是什么原因


报错如下:

Traceback (most recent call last):

  File "mlu_forward.py", line 337, in <module>

    quantification( opt.img, mlu220 = save_offline_model)

  File "mlu_forward.py", line 194, in quantification

    calibration(model, "./coco-eval",20)

  File "mlu_forward.py", line 157, in calibration

    model(preprocess(str(p), keep_aspect_ratio= False))

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/yolov5_v6/models/yolo.py", line 187, in forward

    return self._forward_once(x, profile, visualize)  # single-scale inference, train

  File "/yolov5_v6/models/yolo.py", line 210, in _forward_once

    x = m(x)  # run

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/yolov5_v6/models/common.py", line 97, in forward_fuse

    return self.act(self.conv(x))

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

TypeError: forward() takes 1 positional argument but 2 were given


版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭