jefflai108 / Contrastive-Predictive-Coding-PyTorch

Contrastive Predictive Coding for Automatic Speaker Verification
MIT License
472 stars 96 forks source link

why torch.diag in nce loss? #11

Closed artificertxj1 closed 4 years ago

artificertxj1 commented 4 years ago

NVM, I mis-read the equation. You are right.