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 callbacks to train function in W&B sweep #4742

Merged
merged 2 commits into from
Sep 10, 2021

Conversation

jveitchmichaelis
Copy link
Contributor

@jveitchmichaelis jveitchmichaelis commented Sep 10, 2021

Fix following #4688 which modified the function signature to train

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improvement to the Weights & Biases (WandB) Sweep Functionality in YOLOv5.

📊 Key Changes

  • Added an import of the Callbacks class in the WandB sweep script.
  • Modified the train() function call to include a new callbacks parameter.

🎯 Purpose & Impact

  • The inclusion of Callbacks allows for more customizable training processes by triggering functions at specific stages.
  • This update empowers users to integrate advanced training behaviors and custom actions without modifying the core training logic.
  • Users leveraging WandB sweeps can now benefit from callbacks, potentially leading to better hyperparameter tuning and more robust model training. 🚀

Fix following ultralytics#4688 which modified the function signature to `train`
Copy link
Contributor

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👋 Hello @jveitchmichaelis, thank you for submitting a 🚀 PR! To allow your work to be integrated as seamlessly as possible, we advise you to:

  • ✅ Verify your PR is up-to-date with origin/master. If your PR is behind origin/master an automatic GitHub actions rebase may be attempted by including the /rebase command in a comment body, or by running the following code, replacing 'feature' with the name of your local branch:
git remote add upstream https://github.com/ultralytics/yolov5.git
git fetch upstream
git checkout feature  # <----- replace 'feature' with local branch name
git rebase upstream/master
git push -u origin -f
  • ✅ Verify all Continuous Integration (CI) checks are passing.
  • ✅ Reduce changes to the absolute minimum required for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." -Bruce Lee

@glenn-jocher glenn-jocher changed the title add callbacks to train function in wandb sweep Add callbacks to train function in wandb sweep Sep 10, 2021
@glenn-jocher glenn-jocher changed the title Add callbacks to train function in wandb sweep Add callbacks to train function in wandb sweep Sep 10, 2021
@glenn-jocher glenn-jocher changed the title Add callbacks to train function in wandb sweep Add callbacks to train function in W&B sweep Sep 10, 2021
@glenn-jocher glenn-jocher merged commit 19e28e3 into ultralytics:master Sep 10, 2021
@glenn-jocher
Copy link
Member

@jveitchmichaelis PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐

CesarBazanAV pushed a commit to CesarBazanAV/yolov5 that referenced this pull request Sep 29, 2021
* add callbacks to train function in wandb sweep

Fix following ultralytics#4688 which modified the function signature to `train`

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* add callbacks to train function in wandb sweep

Fix following ultralytics#4688 which modified the function signature to `train`

* Cleanup

Co-authored-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.

2 participants