Skip to content

Running LSTM parser on GPU #46

@HGrant15

Description

@HGrant15

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions