如何使用Cudnn LSTMP？

Question

通常我使用tf.contrib.cudnn_rnn.CudnnLSTM (TensorFlow1.15)，现在我想使用LSTMP(LSTM projection)。我知道tf.contrib.rnn.LSTMCell有LSTMP，但是额外的培训时间对我来说太长了。我发现了那个nvidia support LSTMP。如何在tensorflow(或其他ML工具包)上使用它？

谢谢!

Answer 1

您也可以选择使用tf.contrib.cudnn_rnn.CudnnParamsFormatConverterLSTM，CudnnParamsFormatConverterLSTM是在Cudnn和TF的参数之间进行转换的帮助器类。还可以使用num_proj设置投影矩阵的输出维数。

以下是TensorFlow 1.15中接口的详细文档的link。