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

Designed and maintained by @bonlime and @zakajd

Star the project if you like it!

Project details

Release history Release notifications | RSS feed

0.5.7

Jan 25, 2024

0.5.6

Sep 5, 2023

This version

0.5.5

Aug 29, 2023

0.5.3

Aug 4, 2023

0.5.1

Aug 4, 2023

0.1.9

Jan 22, 2021

0.1.8

Sep 4, 2020

0.1.7

Jul 13, 2020

0.1.5

Jul 8, 2020

0.1.4

May 7, 2020

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.5.5.tar.gz (787.5 kB view details)

Uploaded Aug 29, 2023 Source

Built Distribution

pytorch_tools-0.5.5-py3-none-any.whl (168.2 kB view details)

Uploaded Aug 29, 2023 Python 3

File details

Details for the file pytorch_tools-0.5.5.tar.gz.

File metadata

Download URL: pytorch_tools-0.5.5.tar.gz
Upload date: Aug 29, 2023
Size: 787.5 kB
Tags: Source
Uploaded using Trusted Publishing? No
Uploaded via: twine/4.0.2 CPython/3.10.10

File hashes

Hashes for pytorch_tools-0.5.5.tar.gz
Algorithm	Hash digest
SHA256	`318af7f48d6034ca527cde0e02b4e27fb43cd405152e6aea29cb59fb6f7cc96e`
MD5	`75f70c4242a5ce0ab35045cddd0785a9`
BLAKE2b-256	`aeb0693610104e1fb42177625186326bd77b8fa31a78a7e2c718ce12ab5a563f`

See more details on using hashes here.

File details

Details for the file pytorch_tools-0.5.5-py3-none-any.whl.

File metadata

Download URL: pytorch_tools-0.5.5-py3-none-any.whl
Upload date: Aug 29, 2023
Size: 168.2 kB
Tags: Python 3
Uploaded using Trusted Publishing? No
Uploaded via: twine/4.0.2 CPython/3.10.10

File hashes

Hashes for pytorch_tools-0.5.5-py3-none-any.whl
Algorithm	Hash digest
SHA256	`58ef18326cd92f61640bb4c92a02490fb38e78ce94ada7c70bf865b1a3cbadf0`
MD5	`0f4c1870be8959ff2c5e23c643fbb720`
BLAKE2b-256	`d80775a79cf03ce9c6c537753144715e59ccea767268fe83e9ace79d4bb9f265`