Custom Model from Different version of Ultralytics Not Working

MuhammadMoinFaisal / YOLOv8-DeepSORT-Object-Tracking

YOLOv8 Object Tracking using PyTorch, OpenCV and DeepSORT

666 stars 197 forks source link

Custom Model from Different version of Ultralytics Not Working #6

Closed rabinsubedi closed 1 year ago

rabinsubedi commented 1 year ago

Hello,

I am trying to run my custom model in this tracking code. It seems like this code works with ultralytics<=8.0.10. When I use my custom model trained with latest ultralytics, it doesn't work with the lower version of ultralytics (detection and tracking don't work). When I try to train model with lower version of ultralytics (which actually works with this tracking code), the training shows 'nan' loss (picture below). Can anyone please help? Thanks!

MuhammadMoinFaisal commented 1 year ago

Hi @rabinsubedi, please try to train for atleast 10 epochs and if still the nan loss presists then please review your dataset