search

akanimax / BMSG-GAN

[MSG-GAN] Any body can GAN! Highly stable and robust architecture. Requires little to no hyperparameter tuning. Pytorch Implementation
MIT License
630 stars 105 forks source link

Change the number of gpus #8

Closed wcxchunxin closed 5 years ago

wcxchunxin commented 5 years ago

@akanimax.Hi, Nice work,I've implemented your code.It helped me a lot. But,I want to change the number of gpus.I own two Tesla graphics CARDS.Would you please tell me how to modify it! Thanks!

BlindElephants commented 5 years ago

If you're using cuda, the code provided in this repo should automatically be enabling parallel processing on available GPU hardware:

See here:

    if device == th.device("cuda"):
        self.gen = DataParallel(self.gen)
        self.dis = Discriminator(depth, latent_size, use_eql=use_eql, gpu_parallelize=True).to(device)
wcxchunxin commented 5 years ago

Thank you! It works! May you be happy, my sir! ------------------ 原始邮件 ------------------ 发件人: "Jeremy Stewart"notifications@github.com 发送时间: 2019年5月26日(星期天) 晚上8:41 收件人: "akanimax/BMSG-GAN"BMSG-GAN@noreply.github.com; 抄送: "wcxchunxin"1140657162@qq.com;"Author"author@noreply.github.com; 主题: Re: [akanimax/BMSG-GAN] Change the number of gpus (#8)

If you're using cuda, the code provided in this repo should automatically be enabling parallel processing on available GPU hardware:

See here: if device == th.device("cuda"): self.gen = DataParallel(self.gen) self.dis = Discriminator(depth, latent_size, use_eql=use_eql, gpu_parallelize=True).to(device)

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.

akanimax commented 5 years ago

@wcxchunxin, Glad to know that your problem is solved.

@BlindElephants, Thanks a lot for helping @wcxchunxin out. You do seem to have spent a lot of time in this code. :smile:. Thanks again.

Best regards, @akanimax

wcxchunxin commented 5 years ago

Dear sir， Sure，I did spend a lot of time.But,it's a good work and it's worth it.However,I fail to reproduce your results.512X512 images will take ten days with 200 epochs and it takes too long.Thank you for your reply.

Your, Wang Chunxin.

------------------ 原始邮件 ------------------ 发件人: "Animesh Karnewar"notifications@github.com; 发送时间: 2019年5月27日(星期一) 下午5:20 收件人: "akanimax/BMSG-GAN"BMSG-GAN@noreply.github.com; 抄送: "因为有点傻，所以"1140657162@qq.com;"Mention"mention@noreply.github.com; 主题: Re: [akanimax/BMSG-GAN] Change the number of gpus (#8)

@wcxchunxin, Glad to know that your problem is solved.

@BlindElephants, Thanks a lot for helping @wcxchunxin out. You do seem to have spent a lot of time in this code. . Thanks again.

Best regards, @akanimax

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or mute the thread.