Skip to content
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

The MNIST_DCGAN learning is too slow #1

Open
milk-bottle-liyu opened this issue Nov 22, 2017 · 2 comments
Open

The MNIST_DCGAN learning is too slow #1

milk-bottle-liyu opened this issue Nov 22, 2017 · 2 comments

Comments

@milk-bottle-liyu
Copy link

Sir, as u said the avg time for an epoch is around 180s, while on my server, it shows:
[1/20] - ptime: 372.38, loss_d: 0.597, loss_g: 5.759

My environment is:
ubuntu 16.04+cuda8.0+cudnn 6+ pytorch 0.2 +Titan XP

I also set the worker_num for train data loader to 2, so it shouldn't be a problem of IO.
Do u have any idea of what's going wrong , Sir?

@znxlwm
Copy link
Owner

znxlwm commented Nov 23, 2017

Hi,

I don't know exactly why learning time is different.

I ran my code again using these settings - 64 x 64 MNIST 60,000 images, 128 batch size, 'd' = 128 and got a similar result.

Check your data, batch size, and dimension parameter of network ( 'd' value in my network) etc.

@milk-bottle-liyu
Copy link
Author

milk-bottle-liyu commented Dec 18, 2017 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants