flatironinstitute / catvae

Categorical Variational Autoencoders
BSD 3-Clause "New" or "Revised" License
22 stars 3 forks source link

Batch decouple #62

Closed mortonjt closed 3 years ago

mortonjt commented 3 years ago

This gets rid of the tied batch weights, forcing the encoder batch correction to be performed in the latent space