Skip to content
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

PyTorch Hub load directly when possible #2986

Merged
merged 1 commit into from
Apr 30, 2021
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Apr 30, 2021

This should reduce the verbosity and reduce time-to-load when loading official YOLOv5 PyTorch Hub models.

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements to model loading and customizability in YOLOv5's hub interface.

πŸ“Š Key Changes

  • Simplified loading of pretrained models with attempt_load function for standard use cases.
  • Enhanced create function with additional autoshape and verbosity options.
  • Improved error messages for outdated cache scenarios with explicit instruction to use force_reload=True.

🎯 Purpose & Impact

  • The update streamlines the process of loading pretrained models, which could lead to simpler and cleaner user code.
  • Adding the autoshape flag and verbose settings directly in the create function allows users more control over model initialization and logging.
  • Clarified error messages aim to facilitate troubleshooting for users encountering cache issues.
  • πŸ“ˆ For users, these changes may lead to an improved experience when customizing and deploying YOLOv5 models, with potential time savings and reduced complexity in managing model configurations.

@glenn-jocher glenn-jocher merged commit d08575e into master Apr 30, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-4 branch April 30, 2021 12:59
KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
danny-schwartz7 pushed a commit to danny-schwartz7/yolov5 that referenced this pull request May 22, 2021
Lechtr pushed a commit to Lechtr/yolov5 that referenced this pull request Jul 20, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant