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

Update trainloader in late training #5046

Closed
wants to merge 1 commit into from

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Oct 5, 2021

This PR allows for trainloader updating midway through training. In partial fulfillment of feature request #4461

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improvement to the training data loader during the late training phase in the YOLOv5 model.

πŸ“Š Key Changes

  • Conditional logic added to update the training data loader at 90% of the training epochs.
  • A new data loader creation call during the update step with revised parameters for batching, augmentation, and worker processes.

🎯 Purpose & Impact

  • πŸ”„ The purpose of this change is to refresh the training data loader towards the end of the training process, which may help the model to further improve by seeing the data in a slightly different way or more efficiently.
  • βœ… The impact on users could be improved model accuracy and potentially more efficient training in the final epochs, without affecting earlier stages of the training process.

This PR allows for trainloader updating midway through training. In partial fulfillment of feature request #4461
@glenn-jocher glenn-jocher linked an issue Oct 5, 2021 that may be closed by this pull request
@glenn-jocher glenn-jocher self-assigned this Nov 5, 2021
@github-actions
Copy link
Contributor

github-actions bot commented Dec 6, 2021

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions YOLOv5 πŸš€ and Vision AI ⭐.

@github-actions github-actions bot added the Stale label Dec 6, 2021
@github-actions github-actions bot closed this Dec 13, 2021
@glenn-jocher glenn-jocher deleted the test/update_trainloader branch May 24, 2022 23:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Remove mosaic augmention in the last 15 epochs.
1 participant