3d0c71243d9a1d2b3cd342ab21b5bbd98d48354e,pyannote/audio/embedding/models.py,TristouNet,forward,#TristouNet#Any#,116

Before Change

                // initial hidden and cell states
                h = torch.zeros(self.num_directions_, batch_size, hidden_dim)
                c = torch.zeros(self.num_directions_, batch_size, hidden_dim)
                if gpu:
                    h = h.cuda()
                    c = c.cuda()
                hidden = (Variable(h, requires_grad=False),
                          Variable(c, requires_grad=False))

            elif self.rnn == "GRU":
                // initial hidden state
                h = torch.zeros(self.num_directions_, batch_size, hidden_dim)

After Change

            if self.rnn == "LSTM":
                // initial hidden and cell states
                h = torch.zeros(self.num_directions_, batch_size, hidden_dim,
                                device=device, requires_grad=False)
                c = torch.zeros(self.num_directions_, batch_size, hidden_dim,
                                device=device, requires_grad=False)
                hidden = (h, c)

            elif self.rnn == "GRU":
                // initial hidden state
                hidden = torch.zeros(

Instances