what is the suitable iter for my dataset

xingyizhou / CenterNet2

Two-stage CenterNet

Apache License 2.0

1.2k stars 189 forks source link

what is the suitable iter for my dataset #41

Closed Lg955 closed 3 years ago

Lg955 commented 3 years ago

📚 Documentation

I have 10000 images and 5 classes, what is the suitable MAX_ITER for training Centernet2?

xingyizhou commented 3 years ago

Hi, Thank you for your interest. This is an empirical question and I don't have a clear answer. If you have a validation set, you can turn on TEST.EVAL_PERIOD to visualize the validation curve.

Lg955 commented 3 years ago

I have trained it for 6w iters, but I think it is not enough, maybe it need 10w or more. Thank you！