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

Could you add a training file? #6

Closed
wulikai1993 opened this issue Sep 5, 2020 · 1 comment
Closed

Could you add a training file? #6

wulikai1993 opened this issue Sep 5, 2020 · 1 comment

Comments

@wulikai1993
Copy link

And how long time it takes to train a model?
Do you train the model on a single GPU or multi GPUs?

@StanislasBertrand
Copy link
Owner

Hi,

There is no training method provided. I converted the already trained weights and code from
https://github.com/deepinsight/insightface/tree/master/RetinaFace

Although the provided model is a classif tf.keras model, you could try to train it the usual tf.keras way. In order to match the articles method you would probably also need to implement / find a tf.keras implementation of their multi task loss

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