You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am trying to train the Yolov10 model with a more customized trainer, instead of using the one provided by the default library. I'm trying to reproduce the loss calculation for this model using the v10DetectLoss class, but it always gives this same error:
Search before asking
Question
Hello @THU-MIG team,
I am trying to train the Yolov10 model with a more customized trainer, instead of using the one provided by the default library. I'm trying to reproduce the loss calculation for this model using the v10DetectLoss class, but it always gives this same error:
And I think it comes from this part of the code in v8DetectionLoss class, in utils/loss.py:
The code that I'm using is this one, including a custom dataloader and a custom Trainer using pytorch lightining:
dataloader:
Loss function (v10DetecLoss):
Custom Trainer:
Here are the relevant versions of my dependencies:
Thank you very much in advance for any kind of help!
The text was updated successfully, but these errors were encountered: