-
Notifications
You must be signed in to change notification settings - Fork 103
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Change the number of gpus #8
Comments
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) |
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.
|
@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, |
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.
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@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!
The text was updated successfully, but these errors were encountered: