search

felixrosberg / FaceDancer

Other
465 stars 78 forks source link

Is it possible to finetune pretrained model? #43

Closed baejun10 closed 1 year ago

baejun10 commented 1 year ago

I can't find load pretrained model option at train.py Have you ever tried to finetune model?

felixrosberg commented 1 year ago

Hi, I need more information to help out with troubles loading the model.

Kinda, I have made finetunings for a DeepFake competition. Tried to make some refinements. Other than that, nope. :)

baejun10 commented 1 year ago

I want to finetune your pretrained model(FaceDancer_config_c_HQ.h5). So, I need to load h5 model. but, there is no function to load h5 model in train.py. Also, I assume that FaceDancer_config_c_HQ.h5 may not have discriminator but only generator parts. Can you share your whole files?

baejun10 commented 1 year ago

p.s. I wrote codes that load pretrained model to G(Generator variable). And I tried to finetune model with asian faces. It worked well without pretrained Discriminator!

huysieunhan2910 commented 1 year ago

p.s. I wrote codes that load pretrained model to G(Generator variable). And I tried to finetune model with asian faces. It worked well without pretrained Discriminator!

can you share file ? thannks