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 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 -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git
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.7.tar.gz
(747.8 kB
view details)
File details
Details for the file pytorch_tools-0.1.7.tar.gz.
File metadata
- Download URL: pytorch_tools-0.1.7.tar.gz
- Upload date:
- Size: 747.8 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 | 630def8cd4bdad61df17cbb6aa974e7d9b46202f646f739c7e424d39c113b41b |
|
| MD5 | 1709d61ba6b2cc845ff3b56f82542382 |
|
| BLAKE2b-256 | 9aabcb06160f8fe4cb3828fec434b36adc2f97c317df12b6e317e7e8483ed1e3 |
