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 CSV logging to GenericLogger #9128

Merged
merged 1 commit into from
Aug 24, 2022
Merged

Add CSV logging to GenericLogger #9128

merged 1 commit into from
Aug 24, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Aug 24, 2022

Enable CSV logging for Classify training.

Signed-off-by: Glenn Jocher glenn.jocher@ultralytics.com

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced logging capabilities in YOLOv5 with CSV logger additions and parameter updates.

πŸ“Š Key Changes

  • Transitioned save_dir to use the Pathlib module, enhancing file path handling.
  • Introduced a CSV file logger for metrics, improving persistent and structured data storage.
  • Enabled logging of updated hyperparameters during WandB runs.
  • Streamlined the metric logging function for clarity and consistency across different loggers.

🎯 Purpose & Impact

  • The use of Path ensures better cross-platform compatibility for file paths. 🌍
  • CSV logging allows users to easily track model performance over time in a commonly used format, which is useful for analysis and reporting. πŸ“‰
  • The ability to update parameters in WandB provides better experiment tracking and reproducibility, especially when tuning hyperparameters. πŸ”§
  • Simplifying the log_metrics method reduces code redundancy and enhances maintainability. πŸ› οΈ

Overall, these changes aim to make the logging more robust, user-friendly, and conducive to comprehensive performance tracking.

Enable CSV logging for Classify training.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher self-assigned this Aug 24, 2022
@glenn-jocher
Copy link
Member Author

Looks good!

                  epoch,             train/loss,               val/loss,  metrics/accuracy_top1,  metrics/accuracy_top5,                   lr/0
                      0,                0.86282,                0.85036,                0.87287,                0.98497,                0.00067
                      1,                0.61298,                 0.6996,                0.92968,                0.99338,                0.00034

@glenn-jocher glenn-jocher merged commit f0e5a60 into master Aug 24, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch August 24, 2022 10:31
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Enable CSV logging for Classify training.

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.

None yet

1 participant