How to train the whole framework?

OpenTalker / StyleHEAT

[ECCV 2022] StyleHEAT: A framework for high-resolution editable talking face generation

MIT License

627 stars 77 forks source link

How to train the whole framework? #14

Closed Howx99 closed 2 years ago

Howx99 commented 2 years ago

Hi, thanks for your nice work. I notice that in README, we can train the whole network by running bash bash/train_video_styleheat.sh, but I could not find this file. Also, in utils/trainer.py, a config file(maybe video_styleheat_trainer.yaml ?) is needed to get the model and the trainer. Can you release the training code or the config file for training?

FeiiYin commented 2 years ago

Thanks for your attention. I released the training part code, which is not that clean. You need to modify several path parameters to adapt to your local environment.

Howx99 commented 2 years ago

Ok, thanks a lot!