-
Notifications
You must be signed in to change notification settings - Fork 19
Open
Description
Hi,
I am trying to run your parser over the trained model provided. I would like to run in on my GPU, however it always runs on the CPU.
Do I need to make any changes to the build.xml to compile the project with the tensorflow.jar and libtensorflow.so files?
I noticed in the codebase that the GPUEvaluation.java makes use of TensorFlowInputReader which the rest of the codebase does not use. Is this class the only part of the codebase that makes use of the GPU?
I realise this would have been more appropriate as an email question rather than an 'issue' but I can not close this topic.
Thanks.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels