ottokart / punctuator2

A bidirectional recurrent neural network model with attention mechanism for restoring missing punctuation in unsegmented text
http://bark.phon.ioc.ee/punctuator
MIT License
657 stars 195 forks source link

Multicore CPU Support? #63

Open chrisspen opened 4 years ago

chrisspen commented 4 years ago

Is there any way to optimize training for systems without a GPU?

I'm attempting to train on a system with 8 cores, but no GPU, and I notice it's only using a single core. I didn't expect it to be super efficient without a GPU, but I thought it would at least take advantage of all the cores available.

ottokart commented 3 years ago

Punctuator uses Theano and it should support multiple cores to some extent http://deeplearning.net/software/theano/tutorial/multi_cores.html# , but I personally haven't experimented with training on CPU

ottokart commented 3 years ago

You can also check if you have better luck using Tensorflow version: https://github.com/cadia-lvl/punctuation-prediction/tree/master/punctuator2tf2