Basic framework for training models with 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.
pip install boilr
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:
Install requirements and run the example:
pip install -r requirements.txt CUDA_VISIBLE_DEVICES=0 python example.py
tensorboard 2.1.0(it also works without, but it won't save tensorboard logs)
Release history Release notifications
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|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|