flax-community
/

transformer-vae

Model card Files Files and versions

transformer-vae / model

Commit History

working saving & loading

7633929

Fraser commited on Jul 12, 2021

almost working training

c0f0a4a

Fraser commited on Jul 12, 2021

small fixes

9a1e44d

Fraser commited on Jul 11, 2021

fix enc attention mask

11030c1

Fraser commited on Jul 11, 2021

.

c377535

Fraser commited on Jul 10, 2021

add latent dim args

9416dd9

Fraser commited on Jul 10, 2021

passing most tests

74bc6c7

Fraser commited on Jul 10, 2021

ALMOST WORKING

92871f7

Fraser commited on Jul 9, 2021

small fixes

6f4a0d9

Fraser commited on Jul 9, 2021

shift decoder tokens right

1df308b

Fraser commited on Jul 7, 2021

fix

9c2b63e

Fraser commited on Jul 7, 2021

will move to training args

d661afd

Fraser commited on Jul 7, 2021

fix tabbing

568b60a

Fraser commited on Jul 7, 2021

closer to working

a516fcd

Fraser commited on Jul 7, 2021

getting closer with the code

e6109b4

Fraser commited on Jul 6, 2021

fixes

7bd06a8

Fraser commited on Jul 5, 2021

add todos

db1be99

Fraser commited on Jul 1, 2021

basic code

99674a9

Fraser commited on Jul 1, 2021