Как использовать Cudnn LSTMP?

Question

Обычно я использую tf.contrib.cudnn_rnn.CudnnLSTM (тензор потока 1.15) и сейчас хочу использовать LSTMP (проекция LSTM). Я знаю, что tf.contrib.rnn.LSTMCell имеет LSTMP, но это дополнительное время для обучения слишком длинное для меня. И я обнаружил, что nvidia поддерживает LSTMP . Как использовать его в tenorflow (или другом инструменте ML)?

Спасибо!

Answer 1

Вы также можете использовать tf.contrib.cudnn_rnn.CudnnParamsFormatConverterLSTM, который CudnnParamsFormatConverterLSTM является вспомогательным классом, который преобразует между параметрами Cudnn и TF LSTM. Вы также можете установить выходную размерность для матриц проекции, используя num_proj.
Вот ссылка для подробной документации API в TensorFlow 1.15.