PyTorch extensions for fast R&D prototyping and Kaggle farming
Project description
Pytorch-toolbelt
A pytorch-toolbelt
is a Python library with a set of bells and whistles for PyTorch for fast R&D prototyping and Kaggle farming:
- Easy model building using flexible encoder-decoder architecture.
- Modules: CoordConv, SCSE, Hypercolumn, Depthwise separable convolution and more
- GPU-friendly test-time augmentation
- GPU-friendly inference on huge (5000x5000) images
- Every-day common routines (fix/restore random seed, filesystem utils)
- Fancy losses: Focal, Lovasz, Jaccard and Dice losses
Quick start
TODO: Implement
Installation
TODO: Implement
Documentation
TODO: Implement
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pytorch_toolbelt-0.0.3.tar.gz
(40.7 kB
view hashes)