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

How many steps to train ? #2

Open
mohammedayub44 opened this issue Mar 30, 2021 · 2 comments
Open

How many steps to train ? #2

mohammedayub44 opened this issue Mar 30, 2021 · 2 comments

Comments

@mohammedayub44
Copy link

Hi @solaris33,
I'm using your setup to train on the ICDAR dataset link you have provided. I ran it to about a 1000 steps initially to see if I get any bounding boxes. The loss is not decreasing so much, I'm I doing something wrong here or 1000 steps is too less
When run the eval on 1000'th checkpoint I don't get any boxes with the default thresholds in place, when I reduce the score_map_thresh =0.5 I start to get too many irrelevant boxes.

So my question is how many steps do I have to train to eventually start seeing good eval results with the default thresholds in place ?

Thanks !

@solaris33
Copy link
Owner

Hi.

I set default train step 100,000 (https://github.com/solaris33/EAST-tf2/blob/master/train.py#L17) and use it for example images(https://github.com/solaris33/EAST-tf2/tree/master/examples)

I think at least 50,000 step need to relevant prediction result and loss is meaningless signal (it doesn't decrease after amount of step)

Thanks!

@rahulhanotDTU
Copy link

Does 50000 step are enough for this model to perform the best and i have just started the training process for start

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

No branches or pull requests

3 participants