search

HaloTrouvaille / YOLO-Multi-Backbones-Attention

Model Compression—YOLOv3 with multi lightweight backbones(ShuffleNetV2 HuaWei GhostNet), attention, prune and quantization
https://github.com/HaloTrouvaille/YOLO-Multi-Backbones-Attention
493 stars 118 forks source link

Train erro #15

Open xxd1999 opened 4 years ago

xxd1999 commented 4 years ago

您好，我在用yolov3-tiny训练自己的数据集时出现了以下错误，能帮忙看下么 RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 40 and 48 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71

HaloTrouvaille commented 3 years ago

您好 如果您想用注意力机制 可以加我微信

joe660 commented 3 years ago

您好 如果您想用注意力机制 可以加我微信

Add VoVNet as the backbone. The result is excellent. 这个在哪可以看到？ 另外微信同意一下呀！谢谢

1246886181 commented 3 years ago

您好，我在用yolov3-微小训练自己的数据集时出现了以下错误，能帮忙看下么 RuntimeError：无效参数0：张量的大小必须除尺寸1. GOT 40和48的尺寸在2 /匹配pytorch / aten / src / THC / generic / THCTensorMath.cu：71

解决了吗

yult0821 commented 3 years ago

您好，我在用yolov3-微小训练自己的数据集时出现了以下错误，能帮忙看下么 RuntimeError：无效参数0：张量的大小必须除尺寸1. GOT 40和48的尺寸在2 /匹配pytorch / aten / src / THC / generic / THCTensorMath.cu：71

解决了吗

请问你们解决这个问题了吗？