2016-07-09 14 views
0

私はTF変数の再利用に非常に困惑しています。メソッドのRNNのために、私はこのコード行を見つけることができるよ:Tensorflow変数rnnモジュールでの再利用

if time > 0: vs.get_variable_scope().reuse_variables() 

しかし、dynamic_rnn(私が使用する必要があるメソッド)のために、私はコードのいずれかのreuse_variableラインを見つける、または= Trueの再利用しないでください。 rnn_cellsモジュールで

すべてのRNN細胞は、変数が作成されたかどうチェックしませんが、変数の名前がgraph_keyかに存在する場合LSTM_cellで、_get_concat_variableがうまくチェックしない_linear方法を、使用して初期化されています。

dynamic_rnnは変数を再利用しませんか?変数が作成されているかどうかを明示的にチェックし、変数があればそれを返すメソッドを記述する必要がありますか?

答えて

0

dynamic_rnn関数には、scopeというパラメータがあります。だからwith tf.variable_scope('scope_name', reuse=True)を使って独自のスコープを作成し、dynamic_rnn関数を呼び出すときに設定する必要があります。

model_with_bucketsの実装をチェックアウトすることができます。これは、各バケットに同じモデルを再利用します。

関連する問題