glample / tagger

Named Entity Recognition Tool

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Cuda and Theano Version

nayakt opened this issue · comments

Can you please tell which version of Cuda, CuDNN, and Theano is needed to run your code?

So far I know, the original author, @glample run the code in a cpu.

I am able to run it in GPU, but looks like it is running with batch size 1. Is there a easy way to increase the batch size?

Unfortunately no, batch training was never implemented. The model was already converging fast enough on a CPU for the CoNLL datasets. Also, in my experiments the model was running faster on CPU than on GPU.

Ok. Thanks !!!