Tool box for PyTorch for fast prototyping.
- FitWrapper - Keras like model trainer
- Losses - collection of different Loss functions.
- Metrics - collection of metrics.
- Models - classification model zoo.
- Optimizers
- Segmentation Models - segmentation models zoo
- TTA wrapper - wrapper for easy test-time augmentation
Requires GPU drivers and CUDA already installed.
pip install git+https://github.com/bonlime/pytorch-tools.git@master
It is also recommended to install NVIDIA Apex to allow usage of additional optimizers
pip install ---upgrade -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git
Designed and maintained by @bonlime and @zakajd
Star the project if you like it!