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

W&B artifacts feature addition #1712

Merged
merged 2 commits into from
Dec 17, 2020
Merged

W&B artifacts feature addition #1712

merged 2 commits into from
Dec 17, 2020

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Dec 16, 2020

This PR adds W&B artifact upload ability with the --log-artifacts flag in train.py. When this flag is used final trained model (best.pt if available, else last.pt if --nosave is used) will be uploaded to the W&B cloud console.

$ python train.py --log-artifacts

Screen Shot 2020-12-16 at 2 16 01 PM

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced model saving and logging for YOLOv5 training.

πŸ“Š Key Changes

  • Simplified decision logic for choosing between 'best' or 'last' saved models.
  • Centralized uploading of the final model to a storage bucket (e.g., Google Cloud Storage).
  • Added conditional logging of model artifacts on Weights & Biases (W&B).
  • Ensured the final model is used for post-training evaluation.

🎯 Purpose & Impact

  • Code Clarity: Condensed and cleaned up the code dealing with model saving.
  • Efficiency: Avoids redundant uploads by choosing only the final model for cloud storage.
  • Integration: Provides better integration with W&B by offering artifact logging.
  • Consistency: Guarantees the usage of the same final model for subsequent evaluation, ensuring consistent results.

The update streamlines the training end process, making it clearer, more efficient, and better integrated with external tools like W&B, enhancing the user experience for developers leveraging the YOLOv5 training pipeline. πŸš€

@glenn-jocher
Copy link
Member Author

@AyushExel I've created an artifacts upload feature addition with W&B!

@glenn-jocher glenn-jocher merged commit c923fbf into master Dec 17, 2020
@glenn-jocher glenn-jocher deleted the artifacts branch December 17, 2020 01:52
@AyushExel
Copy link
Contributor

@glenn-jocher looks good!

KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
taicaile pushed a commit to taicaile/yolov5 that referenced this pull request Oct 12, 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

2 participants