mnist

mnist from scratch with pytorch and einops. Kind of serves as a tutorial for building a simple MLP.

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
.gitignore		.gitignore
README.md		README.md
dataset_utils.py		dataset_utils.py
model.py		model.py
requirements.txt		requirements.txt
train.py		train.py

Navigation Menu