Skip to content

Add callbacks to train function in W&B sweep#4742

Merged
glenn-jocher merged 2 commits into
ultralytics:masterfrom
jveitchmichaelis:patch-1
Sep 10, 2021
Merged

Add callbacks to train function in W&B sweep#4742
glenn-jocher merged 2 commits into
ultralytics:masterfrom
jveitchmichaelis:patch-1

Conversation

@jveitchmichaelis

@jveitchmichaelis jveitchmichaelis commented Sep 10, 2021

Copy link
Copy Markdown
Contributor

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`

@github-actions github-actions Bot left a comment

Copy link
Copy Markdown
Contributor

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
Copy Markdown
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>
@glenn-jocher glenn-jocher mentioned this pull request Oct 12, 2021
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