Instruction

This repository is about Auto Encoder implementation in PyTorch

The dataset using is the MNIST dataset.

Files

data.py       --> where processing input dataset
feature_plot.py   --> It's plot the encoder features
inference.py    --> Inference
model_design.py   --> Where the model is
svm_feature.py   --> Plot the features using svc
train.py       --> Training script