Skip to content

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.

2 participants