Hi,
Since the CouplingTransform class in nflows.transforms.coupling module only supports a 1-d mask that splits data along the channel dimension, I am wondering how I would go about implementing the alternating checkerboard mask in the RealNVP paper? Should I used the MaskedAffineAutoregressiveTransform in the nflows.transforms.autoregressive instead? Or are there some other methods or classes that I am not aware of?
Also, thank you so much for providing such a clean implementation of flow models in PyTorch!
Sincerely,
Donglin Wang
Hi,
Since the
CouplingTransformclass innflows.transforms.couplingmodule only supports a 1-d mask that splits data along the channel dimension, I am wondering how I would go about implementing the alternating checkerboard mask in the RealNVP paper? Should I used theMaskedAffineAutoregressiveTransformin thenflows.transforms.autoregressiveinstead? Or are there some other methods or classes that I am not aware of?Also, thank you so much for providing such a clean implementation of flow models in PyTorch!
Sincerely,
Donglin Wang