Skip to main content

Basic framework for training models with PyTorch

Project description

boiler-pytorch

Basic framework for training stuff in PyTorch. It's quite tailored to projects I've been working on lately, so it's meant for personal use. Its sole purpose is to do away with boilerplate code, and having it here makes it easier to share it across projects.

Install

pip install boilr

Usage example/template

There's a usage example that can be useful as template. It's a basic VAE for MNIST quickly hacked together. The example files/folders are:

  • example.py
  • models/
  • experiments/

Install requirements and run the example:

pip install -r requirements.txt
CUDA_VISIBLE_DEVICES=0 python example.py

Tested with:

  • python 3.7.6
  • numpy 1.18.1
  • matplotlib 3.1.2
  • torch 1.4.0
  • torchvision 0.5.0
  • tensorboard 2.1.0 (it also works without, but it won't save tensorboard logs)
  • pillow 7.0
  • tqdm 4.41.1

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for boilr, version 0.6.0
Filename, size File type Python version Upload date Hashes
Filename, size boilr-0.6.0-py3-none-any.whl (20.5 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size boilr-0.6.0.tar.gz (16.9 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page