Backbone for PyTorch training loop
Project description
Backbone for PyTorch training loop
Will try to keep it minimalistic.
pip install back
from back import Bone
Features
- Progress bar
- Checkpoints saving/loading
- Schedulers
- Tensorboard logging
- Early stopping
- Multiple GPUs
TODO
- Learning Rate Finder
- Notebook example
- Docs/tests
Docker
docker build -t backbone_pytorch .
or
sudo docker pull digitman/backbone_pytorch:latest
Examples
SRM
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
back-0.1.0.tar.gz
(3.9 kB
view hashes)