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

About PCB and Duke rank-1 accurate... #39

Closed
zengqixun123 opened this issue May 27, 2018 · 2 comments
Closed

About PCB and Duke rank-1 accurate... #39

zengqixun123 opened this issue May 27, 2018 · 2 comments

Comments

@zengqixun123
Copy link

When I use PCB to train the model ,it always occurs error "out of memory",my GPU is 1080.How much memory do I need to train the model?or change the batchsize?

And when I train the Duke dataset, the train loss and the train accurate is quite small .But the final rank-1 is about 64%as you account.Does it mean overfitting ? what should I do if I want to increase the rank-1 in Duke dataset?

Thanks a lot!

@layumi
Copy link
Owner

layumi commented May 27, 2018

Hi @zengqixun123
Thanks for your attention

  1. Actually I use P40 (16G Memory) for fast training. You may try using batchsize=16 or 8 (You may also need to chose a smaller learning rate).
  2. As for DukeMTMC-reID, you may check the issue Input transformation #32

@zengqixun123
Copy link
Author

Thanks a lot ,it get correct result about 77% rank-1 and 59 map

@layumi layumi closed this as completed Sep 25, 2018
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