PyTorch boilerplate

Heavily based on Weights & Biases transfer learning demo

Requirements:

• Python 3
• Poetry

Quickstart

Download and unzip the dataset in the root folder. The default settings are expecting a directory called pest_classification

poetry install
python run ./pytorch_boilerplate/main.py