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

Batch size in train.py #52

Open
MatiasSaez opened this issue May 18, 2020 · 1 comment
Open

Batch size in train.py #52

MatiasSaez opened this issue May 18, 2020 · 1 comment

Comments

@MatiasSaez
Copy link

MatiasSaez commented May 18, 2020

Hello, first of all I want to say this is the most useful tutorial about object detection using SSD frameworks i've red so far.
In line 20 of train.py a batch_size =8 is defined and then it is passed to the DataLoader in line 71. But, later, in lines 78 and 79, there is a batch size of 32 hardcoded in the formula. Shouldn't be used here the variable batch_size or am I missing something?
Thanks in advance!

@toomy0toons
Copy link

later, in lines 78 and 79, there is a batch size of 32 hardcoded
the batch size above is used to calculate total training epoch, becuase the training epoch of the original paper is 120000 iterations / batch_size(32)

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

2 participants