flax-community
/

transformer-vae

Model card Files Files and versions

transformer-vae / tests

Commit History

working saving & loading

7633929

Fraser commited on Jul 12, 2021

fix reg loss

4e11844

Fraser commited on Jul 12, 2021

almost working training

c0f0a4a

Fraser commited on Jul 12, 2021

fix enc attention mask

11030c1

Fraser commited on Jul 11, 2021

passing most tests

74bc6c7

Fraser commited on Jul 10, 2021