search

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

There seems to be a problem with distributed running codeWhen I entered the training command, the program did not respond #35

Open pfeducode opened 1 year ago

pfeducode commented 1 year ago

When I entered the training command, the program did not respond. How can I solve it 11

samsara-ku commented 1 year ago

Hi, @pfeducode! Did you solve this problem?

I also met this problem, but I couldn't solve this problem.

If you solve this problem, will you share the any idea?

pfeducode commented 1 year ago

I deleted the distributed code, and then it can run normally。

pfeducode commented 1 year ago

There seems to be a problem with distributed running code. When I entered the training command, the program did not respond, and I had to delete the distributed code

harlanhong commented 1 year ago

Please use this command line: CUDA_VISIBLE_DEVICES=0,1,2,3 python run_dataparallel.py --config config/vox-adv-256.yaml --device_ids 0,1,2,3 --name DaGAN_voxceleb2_depth --rgbd --batchsize 48 --kp_num 15 --generator DepthAwareGenerator

harlanhong commented 1 year ago

Actually, I also met this problem when I was using another version pytorch. It seems to only work for "1.9.0+cu111".

image
pfeducode commented 1 year ago

Actually, I also met this problem when I was using another version pytorch. It seems to only work for "1.9.0+cu111". image

Okay, I'll try later

VedantDere0104 commented 1 year ago

After removing the distributed code for the generator and discriminator and making device changes in the "model_dataparallel.py" file, I have successfully got it working on a single GPU.