search

deepinsight / insightface

State-of-the-art 2D and 3D Face Analysis Project
https://insightface.ai
22.01k stars 5.26k forks source link

How to fine-tune ArcFace on a new dataset? #2089

Open sunshineatnoon opened 1 year ago

sunshineatnoon commented 1 year ago

Hi, Thanks for this awesome work. I'm trying to use ArcFace on FFHQ. However, it seems there's domain gap so the running mean and variance does not directly fit to FFHQ. Has anyone tried to fine-tune ArcFace on other datasets and how to handle the BatchNorm layers?

I also tried to freeze the BatchNorm layers, which unfortunately leads to NaN.

lamnt-nd994 commented 1 month ago

Have you solved this problem yet?