模型/教程/ rnn / ptb在R1.4行271上失败没有属性RNNParamsSaveable

Question

API contrib.cudnn_rnn.RNNParamsSaveable已在r1.4中弃用，但在tensorflow教程https://www.tensorflow.org/tutorials/recurrent中使用。目前尚不完全清楚如何修改ptb_word_lm.py中的第271-> 276行以使用新的API，对于默认运行，该API可能应该是contrib.cudnn_rnn.CudnnLSTMSaveable。有人可以告诉我需要改变什么吗？

Answer 1

我按如下方式更改代码，然后就可以了：

if self._cell and rnn_params:
    params_saveable = tf.contrib.cudnn_rnn.CudnnLSTMSaveable(
        rnn_params,
        self._cell.num_layers,
        self._cell.num_units,
        self._cell.input_size,
        self._cell.input_mode,
        self._cell.direction,
        scope="Model/RNN")

删除此行＆＃34; tf.add_to_collection（tf.GraphKeys.SAVEABLE_OBJECTS，params_saveable）＆＃34;