search

flatironinstitute / catvae

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

Decoupling batch weights #64

Closed mortonjt closed 3 years ago

mortonjt commented 3 years ago

Before the batch weights were tied. Now they aren't and it seems to be much easier to train.