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

GPU performance question #15

Open
cjmielke opened this issue Jul 19, 2016 · 2 comments
Open

GPU performance question #15

cjmielke opened this issue Jul 19, 2016 · 2 comments

Comments

@cjmielke
Copy link

I noticed that training on CPU-only is much faster than on GPU. With GPU, usage is about 18% according to nvidia-smi.

Is this because it trains for each epoch, and has to send the data to the card? I see the fast-mode in the code, and it seems to be getting activated with the theano backend. I cannot figure out if Ive got something misconfigured.

@farizrahman4u
Copy link
Owner

Well, while gpu can compute things faster, when there is a lot of data loading/unloading cpus get the advantage. qlearning4k is still the fastest implementation of the q learning algorithm in keras.

@cjmielke
Copy link
Author

I did eventually realize that whenever memory_size < batch_size, the speed is insanely faster but the loss goes to zero. Im going to study the code and papers more to understand what is going on here.

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