Closed Dharmanath closed 5 years ago
@Dharmanath
No, there is no such configuration needed from my end
If you can run other keras code on GPU, this should work as well.
Make sure you have tensorflow-gpu
installed
If you want to test it with a smaller dataset, set the number of samples to 100 instead of 500
Additionally, sequence models like LSTMs don't benefit as much from GPU as say CNNs
Also, please look at the other branch. I have made some changes which I will integrate later.
closing due to no response
Is there any special configuration for usage of GPU?