Doubiiu / CodeTalker

[CVPR 2023] CodeTalker: Speech-Driven 3D Facial Animation with Discrete Motion Prior
MIT License
538 stars 58 forks source link

Question with training stage 2 #9

Closed jinny960812 closed 1 year ago

jinny960812 commented 1 year ago

Hi, I tried to train stage 2 on my dataset, but the loss only oscillates and doesn't go down. Should I reduce the learning rate (it is currently 1e-4) ? Or should I reset the weight between loss_motion and loss_reg? How much should the loss go down? Do you have any tips for stage 2 training? Thank you!