Last active
February 5, 2025 01:44
-
-
Save Chris-hughes10/73628b1d8d6fc7d359b3dcbbbb8869d7 to your computer and use it in GitHub Desktop.
EfficientDet Pytorch-lightning with EfficientNet v2 backbone Blog Post.ipynb
Hi @ramdhan1989 Were you able to solve the operands broadcast issue? I am facing a similar error when training the model
@Chris-hughes10
My model is predicting
The model i have trained has 15 classes. What could have possibly gone wrong?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Dear @Chris-hughes10, Thank you for this amazing work.
Have you done any work as to convert the saved model from torch.save() to onnx?
I am asking this question because I got stuck trying to convert the saved model. The conversion script that I am using is this:
and the error I get is:
No matter how many approaches I tried to solve this problem with, they all eventually resulted in the error above. Any help or guidance if you can kindly provide me with is deeply appreciated.