0

こんにちは、双方向の LSTM レイヤーがあります。

class BiDirLSTMInput(Layer):

  def __init__(self):
    self.bidir_lstm = Bidirectional(
                         LSTM(32, return_sequences=True,return_state=True)
                       )


  def call(self, input):
    o, h1,h2, c1,c2 = self.bidir_lstm(input)
    return [h1,h2]

ご覧のとおり、LSTM から非表示の状態を消費しているだけです (セルの状態ではありません)。

それが理由ですか、次の警告が表示されますか:

WARNING:tensorflow:Gradients do not exist for variables for (backward layer):

  1. lstm_cell_2/kernel:0',
  2. lstm_cell_2/recurrent_kernel:0'
  3. lstm_cell_2/bias:0'

これを無視することは論理的に聞こえません。このエラーに対処するにはどうすればよいですか?

4

2 に答える 2