DisCo

This is the code repository for distance correlation (DisCo) for decorrelation in ML training. See [https://arxiv.org/abs/2001.05310] for details.

Disco.py has the implementation for PyTorch, Disco_tf.py for tensorflow.