Training not Converge

[ldz666666]

Hi, thanks for your amazing work I have tried to reproduce the results, and i followed the data-processing procedure in readme.md. However, when i train on Obama dataset and my own dataset, the training loss seems not converge. here are my arguments and command

python train.py \
-s mnt/workspace/talking_head/GaussianTalker/data/ids/Obama \
--model_path /mnt/workspace/talking_head/GaussianTalker/output/Obama \
--configs arguments/64_dim_1_transformer.py

and the loss log seems super weird

the rendered image has no head, which is different from the original paper.

similar results were observed on my own data, any ideas ? The only difference between my env with yours is i use torch Version: 1.12.1 and cuda 11.3

[ldz666666]

Figured it out, close

[jimydavis]

@ldz666666 how did you figure this out? thanks!