color_syncnet_train.py training not running

sahilg06 / EmoGen

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

Creative Commons Zero v1.0 Universal

343 stars 27 forks source link

color_syncnet_train.py training not running #11

Closed ShangWin closed 1 year ago

ShangWin commented 1 year ago

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/

sahilg06 commented 1 year ago

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?