-
Notifications
You must be signed in to change notification settings - Fork 44
Open
Description
使用YOLOv7结合ACmix,出现如下报错:
`Traceback (most recent call last):
File "/home/liu/桌面/zwx/YOLOv7-main/train.py", line 613, in <module>
train(hyp, opt, device, tb_writer)
File "/home/liu/桌面/zwx/YOLOv7-main/train.py", line 415, in train
results, maps, times = test.test(data_dict,
File "/home/liu/桌面/zwx/YOLOv7-main/test.py", line 110, in test
out, train_out = model(img, augment=augment) # inference and training outputs
File "/home/liu/anaconda3/envs/yolo-torch2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/liu/桌面/zwx/YOLOv7-main/models/yolo.py", line 320, in forward
return self.forward_once(x, profile) # single-scale inference, train
File "/home/liu/桌面/zwx/YOLOv7-main/models/yolo.py", line 346, in forward_once
x = m(x) # run
File "/home/liu/anaconda3/envs/yolo-torch2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/liu/桌面/zwx/YOLOv7-main/models/common.py", line 530, in forward
pe = self.conv_p(position(h, w, x.is_cuda))
File "/home/liu/anaconda3/envs/yolo-torch2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/liu/anaconda3/envs/yolo-torch2/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 463, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/home/liu/anaconda3/envs/yolo-torch2/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 459, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Input type (float) and bias type (c10::Half) should be the `same`
Metadata
Metadata
Assignees
Labels
No labels