wav2lip_288x288 icon indicating copy to clipboard operation
wav2lip_288x288 copied to clipboard

Published 20 hours ago • verified publisher icon primepake

train_syncnet_sam.py is not using GPU (RTX 4090)

Open meng-x opened this issue 10 months ago • 1 comments

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?

meng-x avatar Apr 22 '24 22:04 meng-x

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

ashishjamarkattel avatar Apr 27 '24 14:04 ashishjamarkattel