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

Default --data imagenette160 training (fastest) #9033

Merged
merged 4 commits into from
Aug 19, 2022

Conversation

glenn-jocher
Copy link
Member

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

Signed-off-by: Glenn Jocher [email protected]

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Adjustment of default dataset and clarification on epoch argument in training scripts. πŸŽ›οΈ

πŸ“Š Key Changes

  • Changed the default dataset in classify/train.py from cifar100 to imagenette160.
  • Added help description for the epochs argument in both classify/train.py and train.py to indicate it specifies the total training epochs.

🎯 Purpose & Impact

  • 🎯 The update to a more suitable default dataset is likely for improved out-of-the-box training performance or to showcase the capabilities of YOLOv5 with a different dataset type.
  • πŸ“ˆ The additional help text for the epochs argument improves usability by making it clear to users what this parameter controls, potentially reducing confusion and errors when configuring training sessions.

Signed-off-by: Glenn Jocher <[email protected]>
Signed-off-by: Glenn Jocher <[email protected]>
Signed-off-by: Glenn Jocher <[email protected]>
@glenn-jocher glenn-jocher merged commit 781401e into master Aug 19, 2022
@glenn-jocher glenn-jocher deleted the update/train_defaults branch August 19, 2022 13:06
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
* Default --data `imagenette160` training (fastest)

Signed-off-by: Glenn Jocher <[email protected]>

* Update train.py

Signed-off-by: Glenn Jocher <[email protected]>

* Update train.py

Signed-off-by: Glenn Jocher <[email protected]>

* Update train.py

Signed-off-by: Glenn Jocher <[email protected]>

Signed-off-by: Glenn Jocher <[email protected]>
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.

1 participant