Skip to content

Default --data imagenette160 training (fastest)#9033

Merged
glenn-jocher merged 4 commits intomasterfrom
update/train_defaults
Aug 19, 2022
Merged

Default --data imagenette160 training (fastest)#9033
glenn-jocher merged 4 commits intomasterfrom
update/train_defaults

Conversation

@glenn-jocher
Copy link
Copy Markdown
Member

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

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

🛠️ 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 <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@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 <glenn.jocher@ultralytics.com>

* Update train.py

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

* Update train.py

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

* Update train.py

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.

1 participant