This repository stores a simple Python script which trains a Pytorch model (adapted from an example in https://github.com/pytorch/examples/tree/7f7c222b355abd19ba03a7d4ba90f1092973cdbc).

Python 3.

Requires Python libraries torch and torchvision. With pip:

pip install -r requirements.txt

You may using virtualenv to create a self-contained Python environment for this project. In your shell, in the root of the repository, with virtualenv and pip installed, run:

virtualenv venv
source venv/bin/activate
pip install -r requirements.txt

The source command sets up your shell to use the self-contained environment until you close it. Every time you want to use it, run source venv/bin/activate in the directory you created the virtualenv in.

A Nix flake including a devshell is provided for Nix users. nix develop will set up requirements to then follow the virtualenv instructions above. (Your Nix version needs to support flakes.)

python main.py --save-model

(The default epochs has been retuned from 14 to 3 for faster execution by default. Change this by adding an --epochs x option.)