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
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 details)
File details
Details for the file pytorch_tools-0.1.5.tar.gz.
File metadata
- Download URL: pytorch_tools-0.1.5.tar.gz
- Upload date:
- Size: 738.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/45.2.0 requests-toolbelt/0.9.1 tqdm/4.42.1 CPython/3.6.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 2533449a56ac7ae7ec8b76f03d7aad4a05d10b7939dafb3b8360d9a316ee98f1 |
|
| MD5 | 417522d24cfb882115d6e7b2d509a448 |
|
| BLAKE2b-256 | 79eb1c0e9046d386d1ea15cc4addc8842b156c46cfcb04c22819afa6d103304d |
