/cookiecutter-pytorch

A Cookiecutter template for PyTorch Deep Learning projects.

Primary LanguagePythonMIT LicenseMIT

Cookiecutter Pytorch Template

A Cookiecutter template for PyTorch projects.

Table of Contents

Requirements

  • Python >= 3.8
  • PyTorch >= 1.1
  • Tensorboard >= 1.4

Features

  • Clear folder structure which is suitable for many deep learning projects.
  • Runs are configured via .yml files allowing for easy experimentation.
  • Checkpoint saving and resuming.
  • Tensorboard logging

Usage

$ pip install cookiecutter
$ cookiecutter https://github.com/khornlund/cookiecutter-pytorch
$ cd path/to/repo

A template project has now been created! You can run the MNIST example using:

$ conda env create --file environment.yml
$ conda activate <your-project-name>
$ <your-project-name> train

Example Projects

Here are some projects which use this template:

  1. Severstal Steel Defect Detection (Kaggle)
  2. Aptos Blindness Detection (Kaggle)
  3. Understanding Cloud Organization (Kaggle)

Custom Defaults

If you fork this repo, you can modify cookiecutter.json to provide personalised defaults eg. name, email, username, etc.

Acknowledgements

This template was based on PyTorch Template.