APB2FaceV2 icon indicating copy to clipboard operation
APB2FaceV2 copied to clipboard

Retrain and finetune AudioNet

Open niladridutt opened this issue 5 years ago • 6 comments

While training the network (l2face_trainer), are we fine tuning the audio net? Or is the pretrained weights good enough to be used on any video?

niladridutt avatar Jan 29 '21 13:01 niladridutt

@zhangzjn @YongLiuZJU Any information on how to train the Audio net will be greatly appreciated

niladridutt avatar Feb 01 '21 19:02 niladridutt

We recommend fine-tuning the weight on the custom dataset, even if the weight has some generalization capability.

zhangzjn avatar Feb 02 '21 13:02 zhangzjn

Thanks, but in the code I can't see any method to optimize the audio net (netA) here. I can only see the face re-enactor generator and discriminator weights getting updated

niladridutt avatar Feb 02 '21 14:02 niladridutt

You need to modify the code a little bit about the audionet if you want to train on the custom dataset. This link may help.

zhangzjn avatar Feb 03 '21 06:02 zhangzjn

I see. There seems to be some differences between the 2 codes - addition of a latent vector and some changes in the net architecture as well. Is it possible for you to push the code for training the audionet if posssible? It would be a huge help. Thanks.

niladridutt avatar Feb 03 '21 08:02 niladridutt

Sorry, this code will be considered for update at a later date.

zhangzjn avatar Feb 03 '21 08:02 zhangzjn