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 ---upgrade -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.9.tar.gz
(765.1 kB
view details)
File details
Details for the file pytorch_tools-0.1.9.tar.gz.
File metadata
- Download URL: pytorch_tools-0.1.9.tar.gz
- Upload date:
- Size: 765.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/45.2.0 requests-toolbelt/0.9.1 tqdm/4.49.0 CPython/3.6.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 3f7484d002001a0de73f87cb343590eb70ee151a2fb1c2ecb0674cbc8e815dad |
|
| MD5 | 0d2fb6c37a43d4fe9df24e050915f36e |
|
| BLAKE2b-256 | 3c103093413b67d500b1558791606a753151b520c3bd336be39c91746e18e982 |
