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

my trained model cannot achieve the same performance compared with the provided trained model #4

Open
pipipopo opened this issue Aug 7, 2017 · 6 comments

Comments

@pipipopo
Copy link

pipipopo commented Aug 7, 2017

I use the default training setting in the code. For StanfordOnlineProduct dataset I got:
stanford online products mean recall@ 1 : 0.632077
stanford online products mean recall@ 10 : 0.785517
stanford online products mean recall@ 100 : 0.891618
stanford online products mean recall@ 1000 : 0.959900

How can I train a model with comparable performance?

@PkuRainBow
Copy link
Owner

@pipipopo
Copy link
Author

pipipopo commented Aug 7, 2017

I mean training. Test is OK. For the provided model I got
stanford online products mean recall@ 1 : 0.695367
stanford online products mean recall@ 10 : 0.843716
stanford online products mean recall@ 100 : 0.928379
stanford online products mean recall@ 1000 : 0.975735

But when I train by myself, the performance of final model after 60000 iters is:
stanford online products mean recall@ 1 : 0.632077
stanford online products mean recall@ 10 : 0.785517
stanford online products mean recall@ 100 : 0.891618
stanford online products mean recall@ 1000 : 0.959900

So a gap exsits. I use the default solver prototxt and batch size...

@PkuRainBow
Copy link
Owner

@pipipopo Get it. Try to change the hard_ratio from {1.0, 0.5, 0.2} to {0.5, 0.2, 0.1}. Also please pay attention to the sampling methods. You need to sample 2 big classes firstly and then sample small classes, which is detailed in https://github.com/PkuRainBow/Hard-Aware-Deeply-Cascaded-Embedding_release/blob/master/src_code/sample_stanford_products.py

@zimenglan-sysu-512
Copy link

hi @PkuRainBow,
why try to change the hard_ratio?

@zhengxiawu
Copy link

@pipipopo I trained on CARS, have the same question...

@PkuRainBow
Copy link
Owner

@zhengxiawu Please try hard ratio settings : {0.5, 0.2, 0.1}, which can mine better hard examples.

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

4 participants