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

why early stopping occurs, how to cancel early stopping #4653

Closed
besbesmany opened this issue Sep 2, 2021 · 3 comments · Fixed by #4679
Closed

why early stopping occurs, how to cancel early stopping #4653

besbesmany opened this issue Sep 2, 2021 · 3 comments · Fixed by #4679
Labels
question Further information is requested

Comments

@besbesmany
Copy link

56

@besbesmany besbesmany added the question Further information is requested label Sep 2, 2021
@besbesmany besbesmany changed the title why early stopping occurs, how to continue back why early stopping occurs, how to delete early stopping Sep 2, 2021
@besbesmany besbesmany changed the title why early stopping occurs, how to delete early stopping why early stopping occurs, how to cancel early stopping Sep 2, 2021
@besbesmany
Copy link
Author

also the classification of different classes (map , precision , recall) not appears at the end of epochs as before, how to get this data

@glenn-jocher
Copy link
Member

@besbesmany EarlyStopping stops training when no improvement is seen for patience epochs. By default patience is set to 30, but you can set this to any other value you want, i.e. --patience 100.

yolov5/train.py

Line 465 in fad57c2

parser.add_argument('--patience', type=int, default=30, help='EarlyStopping patience (epochs)')

you can run val.py at anytime to generate full metrics on any model:

python val.py --weights path/to/best.pt --data data.yaml

@glenn-jocher glenn-jocher linked a pull request Sep 5, 2021 that will close this issue
@glenn-jocher
Copy link
Member

@besbesmany good news 😃! Your original issue may now be fixed ✅ in PR #4653. To receive this update:

  • Gitgit pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
  • PyTorch Hub – Force-reload with model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
  • Notebooks – View updated notebooks Open In Colab Open In Kaggle
  • Dockersudo docker pull ultralytics/yolov5:latest to update your image Docker Pulls

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants