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 on_fit_epoch_end callback #5232

Merged
merged 3 commits into from
Oct 18, 2021
Merged

Add on_fit_epoch_end callback #5232

merged 3 commits into from
Oct 18, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Oct 18, 2021

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancements to training callbacks for COCO dataset evaluations and improved results logging.

πŸ“Š Key Changes

  • Added an additional callback condition specifically for the COCO dataset during the training phase.
  • Updated the on_train_end callback to include the results in its parameters.

🎯 Purpose & Impact

  • 🎲 Tailored Callbacks for COCO Dataset: The update enables custom behavior during training when the model is using the COCO dataset, likely to process evaluation results specifically for this common benchmark.
  • πŸ“ˆ Enhanced Results Logging: By including results in the on_train_end callback, more detailed information will be logged at the end of training, providing users with better insights into model performance.
  • Both changes should improve the user's insight and control over the training process, particularly when working with the well-known COCO dataset, potentially leading to more informed decision-making and streamlined workflows.

@glenn-jocher glenn-jocher merged commit 752d97d into master Oct 18, 2021
@glenn-jocher glenn-jocher deleted the update/callback branch October 18, 2021 12:28
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* Add `on_fit_epoch_end` callback

* Add results to train

* Update __init__.py
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