harlanhong / CVPR2022-DaGAN

Official code for CVPR2022 paper: Depth-Aware Generative Adversarial Network for Talking Head Video Generation
https://harlanhong.github.io/publications/dagan.html
Other
957 stars 125 forks source link

请问SPADE_DaGAN_vox_adv_256.pth.tar模型自己怎么训练出来 #50

Open hsk-yjk opened 1 year ago

harlanhong commented 1 year ago

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --master_addr="0.0.0.0" --master_port=12348 run.py --config config/vox-adv-256.yaml --name DaGAN --rgbd --batchsize 12 --kp_num 15 --generator SPADEDepthAwareGenerator

替换参数--generator