Contrastive-Predictive-Coding-PyTorch
Contrastive-Predictive-Coding-PyTorch copied to clipboard
Hey should 'correct' only be if the final prediction was correct? Same for accuracy?
https://github.com/jefflai108/Contrastive-Predictive-Coding-PyTorch/blob/a9dab4e759aaa68dce1b1ada46a8035076ba3296/src/model/model.py#L100
Meaning, should that correct variable be += ?
I noticed that too.
I think you're right, and after the loop over the timesteps, the sum should be divided by batch_size*num_of_timesteps.