Skip to content

This is a reimplementation of the blog post "Building Autoencoders in Keras". Instead of using MNIST, this project uses CIFAR10.

License

Notifications You must be signed in to change notification settings

wcycqjy/PyTorch-CIFAR-10-autoencoder

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

building-autoencoders-in-Pytorch

This is a reimplementation of the blog post "Building Autoencoders in Keras". Instead of using MNIST, this project uses CIFAR10.

Current Results (Trained on Tesla K80 using Google Colab)

First attempt: (BCEloss=~0.57)
decode

Best Predictions so far: (BCEloss=~0.555)
decode decode

Targets:
target target

Previous Results (Trained on GTX1070)

First attempt: (Too much MaxPooling and UpSampling)
decode

Second attempt: (Model architecture is not efficient)
decode

Targets:
decode

License

MIT

About

This is a reimplementation of the blog post "Building Autoencoders in Keras". Instead of using MNIST, this project uses CIFAR10.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%