Implementation of Invertible Neural Networks (INNs) based on Ardizzone et al. (2019) using TensorFlow2 + Keras

less than 1 minute read

Published:


This repository includes the code example of the invertible neural networks (Ardizzone et al., 2019) implemented using TensorFlow2 with Keras. Two complementary coupling layers were implemented and toy examples were provided similar to the paper. The current code is largely based on the original PyTorch implementation by the authors, but simplified for the easier understanding than the original code and for the personal use.