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

Add git info to training checkpoints #9655

Merged
merged 11 commits into from
Nov 19, 2022
Merged

Add git info to training checkpoints #9655

merged 11 commits into from
Nov 19, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 29, 2022

Assist with reproducibility, partially resolve #9643

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced YOLOv5 training logs with git metadata for better version tracking.

πŸ“Š Key Changes

  • Added gitpython Module: A new dependency was added to the requirements.txt file to enable git-related operations within the code.
  • Introduced GIT Variable: A GIT variable was created in utils/general.py to store the repository's remote URL, branch, and latest commit hash.
  • Updated Training Artifacts: Training outputs, such as model checkpoints, now include git information to couple the code version with trained models.

🎯 Purpose & Impact

  • Improved Reproducibility: Including git metadata ensures that training outcomes can be traced back to the exact code version, facilitating error tracking and experiment reproducibility.
  • Enhanced Debugging: Developers and users can more easily relate model performance to specific commits, aiding in debugging and model improvement endeavors.
  • Streamlined Workflow: This addition simplifies identifying the codebase status when sharing or reviewing models, leading to more streamlined collaboration among developers and researchers.

@glenn-jocher glenn-jocher changed the title Add git info to training checkpoints Add git info to training checkpoints Sep 29, 2022
@glenn-jocher glenn-jocher self-assigned this Nov 19, 2022
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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.

How to know which sub version of yolo v5 you are using
1 participant