-
Notifications
You must be signed in to change notification settings - Fork 302
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Pruned Yolov8 model not loading? #363
Comments
have you solove this question? i had met same problem |
@luoshiyong Not yet. How did you saved your model ? |
Hi, I managed to use the example demo of Yolov8. from ultralytics.nn.tasks import attempt_load_one_weight But for that I need to import the C2f_v2 class as it's not part of ultralytics YOLOV8 :
Hope it will help! |
@VainF I have trained a custom YOLOv8 model. After training i have successfully pruned the model.
After I save pruned model. Is it the correct way to save pruned model?
After saving model I load model and it showing the following error:
Also I need to ask is it necessary to train again on a pruned model?
The text was updated successfully, but these errors were encountered: