Running on GPU
andvalenzuela opened this issue · comments
Do you know how to train this model on your own GPU?
you need to install tensorflow1.8.0 and tensorflow-gpu1.8.0, cuda 9.0 in you own computer. it works , i had trained it several time.
shaoanlu / faceswap-GAN
A denoising autoencoder + adversarial losses and attention mechanisms for face swapping.