This project focuses on building and training a neural network to recognize handwritten digits using the MNIST dataset within the PyTorch framework. The goal is to refresh my memory on PyTorch basics and understand how to apply it for practical machine learning tasks.
To get a local copy up and running, follow these simple steps:
-
Clone the repository:
git clone https://github.com/penguyen72/mnist.git
-
Change directory to mnist
cd mnist
-
Switching to Python 3.10
pyenv local 3.10
-
Create virtual environment
python -m venv venv
-
Install dependencies
pip install -r requirements.txt
Here's how to get started with the application
jupyter notebook
Refreshing memory on PyTorch with simple project