sahilg06 / EmoGen

PyTorch Implementation for Paper "Emotionally Enhanced Talking Face Generation" (ICCVW'23 and ACM-MMW'23)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

color_syncnet_train.py training not running

ShangWin opened this issue · comments

commented

can you advise on a workaround for this? I have been waiting for color_syncnet_train.py to run for over a few hours numerous times, and nothing is loading or running yet. you mentioned internal changes about tqdm, but what is the potential issue specifically with it?

I could run emotion_disc_train.py training, but not the color_syncnet_train.py training

code entered in cmd line:
python color_syncnet_train.py --data_root ./preprocessed_dataset/ --checkpoint_dir ./checkpoint/

Hi @ShangWin, I am not sure about the specific issue. Actually, it works for me after a few secs (at max 1-2 minutes). Maybe you can try removing tqdm from the code?