Code question in src/model/model.py#L113

jefflai108 / Contrastive-Predictive-Coding-PyTorch

Contrastive Predictive Coding for Automatic Speaker Verification

MIT License

472 stars 96 forks source link

Code question in src/model/model.py#L113 #2

Closed joeyy5588 closed 5 years ago

joeyy5588 commented 5 years ago

In model.py line 113 : output2, hidden1 = self.gru2(forward_seq, hidden1) perhaps it should be : output2, hidden2 = self.gru2(forward_seq, hidden1) ?

jefflai108 commented 5 years ago

hi, you are correct. my bad.

it should have been output2, hidden2 = self.gru2(forward_seq, hidden2)