Training Interruption at Epoch 54 #325
-
Hello, My model training consistently gets interrupted at epoch 54 on both Google Colab and my local system. Has anyone else faced this issue or have any suggestions for resolving it? Thanks! |
Beta Was this translation helpful? Give feedback.
Answered by
sorohere
Jul 7, 2024
Replies: 1 comment
-
It sounds like your training process might be exceeding your local resources. you could try using smaller pre-trained models of the same type to train on your custom dataset. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
NikhilKalloli
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It sounds like your training process might be exceeding your local resources. you could try using smaller pre-trained models of the same type to train on your custom dataset.
Here's a link to find all such models: https://docs.ultralytics.com/models/yolov10/#model-variants