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

training problems #7

Open
xz-123-new opened this issue Sep 15, 2022 · 1 comment
Open

training problems #7

xz-123-new opened this issue Sep 15, 2022 · 1 comment

Comments

@xz-123-new
Copy link

I got a problem with training this model, looking forward to your generous reply. I train this model on hicodet dataset with resnet-50 as backbone , all the parameters concerning training are set following the paper except batch size, i choose batch size =2 to train because of GPU memory ,after i train 80 epoches and even 150 epoches now, i didn't get the mAP value it should be .For now, i only got ""test_mAP": 0.29916627302565096, "test_mAP rare": 0.23861787027422618, "test_mAP non-rare": 0.3172521595617908 ".Could you please give me some advice about training stradegy to attain the correct number?thanks!

@SherlockHolmes221
Copy link
Owner

The batch size is set as 2 for per GPU by default. I do not understand your meaning.
advice: the learning rate should be changed according to batch size

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