feat: allow scanning - Githubissues

patil-suraj / vit-vqgan

JAX implementation ViT-VQGAN

MIT License

77 stars 11 forks source link

feat: allow scanning #21

Closed borisdayma closed 2 years ago

borisdayma commented 2 years ago

TODO:

[x] update model to allow scanning
[x] allow unscanning
[x] split optimizer per scanned/vanilla layers

borisdayma commented 2 years ago

Already merged. Obsolete.