search

bubbliiiing / yolov8-pytorch

这是一个yolov8-pytorch的仓库，可以用于训练自己的数据集。
GNU General Public License v3.0
675 stars 80 forks source link

怎么load pth #33

Open Arrosw opened 1 year ago

Arrosw commented 1 year ago

model=YOLO('./yolov8s.pt') checkpoint = torch.load('./best_epoch_weights.pth') 报错RuntimeError: Error(s) in loading state_dict for YOLO: 怎么解决呀？求教，感谢！