eeskimez / emotalkingface

The code for the paper "Speech Driven Talking Face Generation from a Single Image and an Emotion Condition"
MIT License
161 stars 29 forks source link

about the training receipy #1

Open flyingfish007 opened 2 years ago

flyingfish007 commented 2 years ago

Thank you very much for open source your code and model. I tried to trained your code. It works fine. I just want to know how many epoch did you train for the pre-train disc emotion and pre-train the generator. Thanks.

eeskimez commented 2 years ago

I appreciate your interest in our work! We pre-train the disc emotion and generator around 100K iterations, and we trained the network with full objective function for another 100K iterations. Note that our networks were trained with a single GTX 1080 TI. With other cards, the batch size and number of iterations might differ.