wywu / ReenactGAN

[ECCV 2018] ReenactGAN: Learning to Reenact Faces via Boundary Transfer
https://wywu.github.io/projects/ReenactGAN/ReenactGAN.html
197 stars 41 forks source link

gpu and CelebV/Donald_Trump #8

Closed jjandnn closed 4 years ago

jjandnn commented 4 years ago

1,建议在read me中提醒大家--gpu_ids 0,1,2,3,这个参数在train_Transformer.sh中必须根据自己的设备情况设定。很多兄弟都习惯先拿colab或kaggle跑一跑,k80和p100只有一块gpu,只能--gpu_ids 0 ,写排序gpu,cuda就会报错。 2,在train_Decoder.sh中,你把CelebV/Donald_Trump写成了CelebV/Donald_trump小写。于是报错,找不到文件夹了。下载的数据集默认是大写。

祝顺利