Zejun-Yang / AniPortrait

AniPortrait: Audio-Driven Synthesis of Photorealistic Portrait Animation
Apache License 2.0
4.65k stars 582 forks source link

About stage1 training steps #182

Open Vincent-luo opened 4 months ago

Vincent-luo commented 4 months ago

I find in the "stage1.yaml" file max_train_steps is set to 300,000, and I want to ask is it really necessary to train such a long schedule? In animate anyone paper, they say just train stage1 for 30,000 steps. And in my own experiment the loss doesn't change too much after 30,000 steps. It would be helpful if you can provide some guidance about how to find a proper training schedule.