jizhishutong / YOLOU

YOLOv3、YOLOv4、YOLOv5、YOLOv5-Lite、YOLOv6-v1、YOLOv6-v2、YOLOv7、YOLOX、YOLOX-Lite、PP-YOLOE、PP-PicoDet-Plus、YOLO-Fastest v2、FastestDet、YOLOv5-SPD、TensorRT、NCNN、Tengine、OpenVINO

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train my dataset? there are some issues when load pretrained weights.

woailuoshenqi opened this issue · comments

commented

size mismatch for model.33.cls_preds.2.bias: copying a param with shape torch.Size([80]) from check point, the shape in current model is torch.Size([14]).

Have you fixed the problem?

commented

NO, not yet.