EasySRL
EasySRL copied to clipboard
Running LSTM parser on GPU
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.
The GPU version can be used with two simple changes:
(1) Use this model instead.
(2) Download this binary and place it in the lib
directory.
An example of a script preparing this (amongst other things) can be found here.
Hope that helps!
java -jar easysrl.jar --model modelFolder --outputFormat ccgbank does not work