Metadata-Version: 1.2
Name: pytorch-tools
Version: 0.1.4
Summary: Tool box for PyTorch
Home-page: https://github.com/bonlime/pytorch-tools
Author: Emil Zakirov
Author-email: bonlimezak@gmail.com
License: MIT License
Description: # Pytorch-Tools
        
        Tool box for PyTorch for fast prototyping.
        
        # Overview  
        * [FitWrapper](./pytorch_tools/fit_wrapper/) - Keras like model trainer
        * [Losses](./pytorch_tools/losses/) - collection of different Loss functions.
        * [Metrics](./pytorch_tools/metrics/) - collection of metrics.
        * [Models](./pytorch_tools/models/) - classification model zoo.
        * [Optimizers](./pytorch_tools/optim/)
        * [Segmentation Models](./pytorch_tools/segmentation_models/) - segmentation models zoo
        * [TTA wrapper](./pytorch_tools/tta_wrapper/) - wrapper for easy test-time augmentation
        
        # Installation
        Requeres 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`
        
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Requires-Python: >=3, <4
