search

shaoanlu / faceswap-GAN

A denoising autoencoder + adversarial losses and attention mechanisms for face swapping.
3.39k stars 841 forks source link

Running on GPU #163

Open andvalenzuela opened 4 years ago

andvalenzuela commented 4 years ago

Do you know how to train this model on your own GPU?

wiki-lai commented 4 years ago

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.