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

Problems with training #87

Open
Szakulli07 opened this issue Oct 22, 2023 · 0 comments
Open

Problems with training #87

Szakulli07 opened this issue Oct 22, 2023 · 0 comments

Comments

@Szakulli07
Copy link

Hello, this is not really issue with dataset but I hope mayby you can help me.

I try to train decent image retrival model on your dataset. However I can not get anything at least near satisfying.

With resnet18 pretrained on image net without any training I get about 2% on validation set when considering top 10 results.

I've already tried using resnet18, resnet34, resnet50, some VITs all pretrained from image net. AdamW as optimizer with lr 1e-3, 1e-4, 1e-5. Triplet loss with margins varying from 1e-2 to 1. Batch hard mining, hard negative mining, easy positive mining and some other kinds of minings. Batch size depending of model from 400 to 1600. Using only shop images, only customer images, using both, training on only validation set.

But all that didn't get me more than 5.7% on this set.

I know I can use some more advanced techniques for pooling, rerankng etc but shouldn't I get at least decent result in this basic setup?

I would be more than grateful for any advice. This is small part of my master thesis but for now it is a lot harder than I supposed.

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

1 participant