Tool box for PyTorch
Project description
Pytorch-Tools
Tool box for PyTorch for fast prototyping.
Overview
- 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
Installation
Requires GPU drivers and CUDA already installed.
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git
pip install git+https://github.com/bonlime/pytorch-tools.git@master
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_tools-0.1.5.tar.gz
(738.1 kB
view hashes)