78197159b897391afd44862381694134bb8d049f,pyannote/audio/labeling/models.py,StackedLSTM,__call__,#StackedLSTM#Any#,63

Before Change

                               dropout_U=0.0)(forward)

                if self.bidirectional:
                    backward = LSTM(name="backward_{i:d}".format(i=i),
                                    output_dim=output_dim,
                                    return_sequences=True,
                                    activation="tanh",
                                    dropout_W=0.0,
                                    dropout_U=0.0)(backward)
            else:
                // first LSTM
                forward = LSTM(name="forward_{i:d}".format(i=i),
                               output_dim=output_dim,

After Change

        n_lstm = len(self.lstm)
        for i, output_dim in enumerate(self.lstm):

            params = {
                "name": "lstm_{i:d}".format(i=i),
                "output_dim": output_dim,
                "return_sequences": True,
                "activation": "tanh"
            }

            // first LSTM needs to be given the input shape
            if i == 0:
                params["input_shape"] = input_shape

Instances