How to speedup rnn training speed of tensorflow?

后端 未结 2 1764
难免孤独
难免孤独 2021-02-09 04:52

Now base tensorflow-char-rnn I start a word-rnn project to predict the next word. But I found that speed is too slow in my train data set. Here is my training details:

2条回答
  •  太阳男子
    2021-02-09 05:38

    One other possible way you can speed up training, and the possible reason for your lack of utilisation of the GPU, is you are using placeholders. You should be using queues, if using Tensorflow < 1.2, and the tf.contrib.data module otherwise.

    https://www.tensorflow.org/programmers_guide/threading_and_queues

提交回复
热议问题