search

primepake / wav2lip_288x288

MIT License
524 stars 135 forks source link

train_syncnet_sam.py is not using GPU (RTX 4090) #138

Open meng-x opened 2 months ago

meng-x commented 2 months ago

Screenshot 2024-04-23 070907 it seems that train_syncnet_sam.py is not using GPU for training the SyncNet. torch successfully detected GPU and actually other program work great with the GPU on the same PC. Screenshot 2024-04-23 071206 using Python 3.7 on WSL2(Ubuntu 22) on Windows 11. using 3s long video clips for training. using syncnet_batch_size=64 (default) -> shall I make this a larger number to make use of larger GRAM or something?

ashishjamarkattel commented 2 months ago

You can load the model in GPU. in the code in train_syncnet.py