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.2.tar.gz
(4.7 kB
view hashes)
Built Distribution
back-0.1.2-py3-none-any.whl
(5.2 kB
view hashes)