This repository describes how to build, train and test deep learning networks in pytorch. New models can be added to the models directory and loaded in main.py.
Create a virtual environment using venv
python3 -m venv envSource the virtual environment
source env/bin/activateUse the package manager pip to install requirements.
pip install -r requirements.txtpython main.py --mode=[Train/Test] --resume=[True/False]There are more flags that can be found in the main function.