Imported from https://github.com/jefflai108/Contrastive-Predictive-Coding-PyTorch