YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series
Apache License 2.0
283 stars 39 forks source link

Unable to load pretrained checkpoints for yolov_x and yolov_l #23

Closed vinceyoung closed 1 year ago

vinceyoung commented 1 year ago

Dear Author,

Thanks for sharing your great work. I'm trying to train yolov_l and yolov_x, but it looks like it's unable to load the checkpoints for yoloxl_vid.pth and yoloxx_vid.pth provided in google drive.

Here's my training command python tools/vid_train.py -f exps/yolov/yolov_l.py -c pretrained_weights/yoloxl_vid.pth --fp16 python tools/vid_train.py -f exps/yolov/yolov_x.py -c pretrained_weights/yoloxx_vid.pth --fp16

And I'm getting this error

image

My command below for training yolov_s is working fine. Is it possible that the large and extra large checkpoints got corrupted? python tools/vid_train.py -f exps/yolov/yolov_s.py -c pretrained_weights/yoloxs_vid.pth --fp16

Thanks for the help!

YuHengsss commented 1 year ago

Dear Author,

Thanks for sharing your great work. I'm trying to train yolov_l and yolov_x, but it looks like it's unable to load the checkpoints for yoloxl_vid.pth and yoloxx_vid.pth provided in google drive.

Here's my training command python tools/vid_train.py -f exps/yolov/yolov_l.py -c pretrained_weights/yoloxl_vid.pth --fp16 python tools/vid_train.py -f exps/yolov/yolov_x.py -c pretrained_weights/yoloxx_vid.pth --fp16

And I'm getting this error

image

My command below for training yolov_s is working fine. Is it possible that the large and extra large checkpoints got corrupted? python tools/vid_train.py -f exps/yolov/yolov_s.py -c pretrained_weights/yoloxs_vid.pth --fp16

Thanks for the help!

Hello, thanks for your interest in our work. We find the same error with the two pretrain-weights and we have updated it, please try it again.

vinceyoung commented 1 year ago

It works for me now as well, thank you!