Deep Learning with Pytorch Essentials

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.

Installation

Create a virtual environment using venv

python3 -m venv env

Source the virtual environment

source env/bin/activate

Use the package manager pip to install requirements.

pip install -r requirements.txt

Usage

python main.py  --mode=[Train/Test] --resume=[True/False]

There are more flags that can be found in the main function.

License

MIT