search

zhanglonghao1992 / One-Shot_Free-View_Neural_Talking_Head_Synthesis

Pytorch implementation of paper "One-Shot Free-View Neural Talking-Head Synthesis for Video Conferencing"
Other
764 stars 143 forks source link

Training becomes ubstable with jacobean #25

Open AlanLu0808 opened 2 years ago

AlanLu0808 commented 2 years ago

It seems jacobean not added in your code. I tried to add jacobean but find the loss become NaN after servel epochs.Any help? thx!

zhanglonghao1992 commented 2 years ago

@AlanLu0808 Just set estimate_jacobian to True in config file to use jacobian. My training is stable with jacobian.