aneesh-joshi / LSTM_POS_Tagger

A simple POS Tagger made using a Bidirectional LSTM using keras trained on the Brown Corpus
34 stars 19 forks source link

GPU configuration #9

Closed Dharmanath closed 5 years ago

Dharmanath commented 6 years ago

Is there any special configuration for usage of GPU?

aneesh-joshi commented 6 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

aneesh-joshi commented 6 years ago

Additionally, sequence models like LSTMs don't benefit as much from GPU as say CNNs

aneesh-joshi commented 6 years ago

Also, please look at the other branch. I have made some changes which I will integrate later.

aneesh-joshi commented 5 years ago

closing due to no response