Skip to content

Latest commit

 

History

History

Autoencoder2D

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

Simple 2D Autoencoder

This is an example of a simple 2-dimensional autoencoder model, using KuzushijiMNIST as a training dataset. It should produce output similar to the following:

Epoch 1

Epoch 10

Setup

To begin, you'll need the latest version of Swift for TensorFlow installed. Make sure you've added the correct version of swift to your path.

To train the model, run:

swift run -c release Autoencoder2D