The --iterations and --coarse_iterations params to train.py don't seem to work

cvlab-kaist / GaussianTalker

Official implementation of “GaussianTalker: Real-Time High-Fidelity Talking Head Synthesis with Audio-Driven 3D Gaussian Splatting” by Kyusun Cho, Joungbin Lee, Heeji Yoon, Yeobin Hong, Jaehoon Ko, Sangjun Ahn and Seungryong Kim

Other

300 stars 36 forks source link

The --iterations and --coarse_iterations params to train.py don't seem to work #25

Closed robmeadows closed 5 months ago

robmeadows commented 5 months ago

I tried setting it like this but it still does 10K iterations. If I go into the code and override op.iterations, I can get it to do more. python train.py -s data/obama --model_path models/obama/ --configs arguments/64_dim_1_transformer.py --iterations 100000 --coarse_iterations 20000

jimydavis commented 5 months ago

check this file: arguments/64_dim_1_transformer.py

kyusuncho commented 5 months ago

As @jimydavis said, the parameters need to be configured inside the configuration file inside the arguments folder.