/mnist-tinygrad

Learning mnist dataset with tinygrad

Primary LanguagePython

Installation Steps

  • Clone this repository.
  • cd to ./mnist-tinygrad.
  • Create a virtual environment using python3 -m venv env.
  • Activate environment using source ./env/bin/activate (it's for Linux and Mac, use .\env\Scripts\activate for Windows).
  • Run which python to make sure that your python interpreter comes from your virtual environment. Use where python for Windows.
  • Install Tinygrad with the official documentation (link).

Usage

Just run one of the following commands to run the model in test or train mode:

  • To train the model: python main.py --train.
  • To test the model: python main.py --test.