Status: Unmaintained Legacy Code. If you want to build upon this, the PyTorch implementation of the loss might be a good starting point.

For a more recent approach to disentangling with up-to-date code, see Network-to-Network Translation with Conditional Invertible Neural Networks (especially Sec. C in the supplementary of the paper).