Skip to content

dhuynh95/fastai_autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

36 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

fastai_autoencoder

Fastai autoencoder is a library around Fastai to make the training of autoencoders easier.

To create an autoencoder learner you need to split your model in three steps, an encoder, a bottleneck, and a decoder, then use AutoEncoderLearner.

To train VAE, a Fastai hook called VAEHook can be used.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published