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

How did you train your pre-trained model? #94

Open
hanbongyoo opened this issue Nov 2, 2021 · 1 comment
Open

How did you train your pre-trained model? #94

hanbongyoo opened this issue Nov 2, 2021 · 1 comment

Comments

@hanbongyoo
Copy link

hanbongyoo commented Nov 2, 2021

Hello~ I'm a beginner in face recognition.

I wanted to train the model myself to get similar performance to your pre-trained model.
But the problem was that training accuracy was always 0 and threshold was 0.99.
So I used Insightface(arcface_torch) code to train the model myself.
However, despite using a larger dataset(ms1mv3), the performance was worse than your model.

I tried to change Insightface model like your model, but the performance was still worse than your model.(e.g. image channel(3->1), resnet18 model structure, batchsize)
I tested lfw dataset and my own dataset(Asian).
I think the cause of the difference in performance is the learning rate and training epoch.
In Insightface, training epoch is 25 and only 3 decay_epoch.

Can you explain how did you train your model in detail?
or what do you think that the reason about can't catch up your model performance?

Thank you for reading.

@jonilaserson
Copy link

Also asking the same question.

I was able to achieve 97.46% accuracy on lfw test using this code after 110 epochs.
However, the pretrained model gets 99.33% accuracy.

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