打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
使用官方文档进行操作,在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
热门帖子
精华帖子