LSTMモデルのパーツパラメータを復元する方法

Question

私はLSTMモデルL1をW-lstm、b-lstm、Wy、byなどのパラメータで事前トレーニング用に訓練しました。

そして、別のWyと、Wy2とby2を仮定して、新しいLSTMモデルL2を訓練したいと思います。

したがって、L1からW-lstmとb-lstmを復元し、Wy2とby2を開始するにはどうすればよいですか？その後、トレーニング。

saver.restore（）？

ありがとうございました。

Answer 1

セーバーを作成するときに、保存する変数のリストを渡すことができます。したがって、最初のLSTMのためだけにセーバを作成することができます。ここに私が意味するものの例があります。 LSTMを変数スコープ内に作成して、そのスコープ内の変数のみを保存するセーバーを作成するだけです（これはブラウザに入力したコードなので、タイプミスがある可能性があります）。

lstm = tf.nn.rnn_cell.BasicLSTMCel(1024) 
with tf.variable_scope('L1') as scope: 
    state = INITIAL_STATE 
    for t in range(TIME_STEPS): 
     output, state = lstm(input, state) 
     scope.reuse_variables() 

... 

l1_saver = tf.train.Saver([v for v in tf.all_variables() if 'L1' in v.name])