/contrastive-predictive-coding

PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)

Primary LanguagePython

Contrastive Predictive Coding

Implementation of PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018).

Paper

Usage

python main.py

Set the experiment's configuration parameters in the config/audio folder's files. This implementation uses sacred to manage experiments.