torchhk 0.80

Custom utils for PyTorch

pytorch-custom-utils

This is a lightweight repository to help PyTorch users.

Usage

:clipboard: Dependencies

• torch 1.4.0
• torchvision 0.5.0
• python 3.6
• matplotlib 2.2.2
• numpy 1.14.3
• seaborn 0.9.0
• sklearn
• plotly

:hammer: Installation

• pip install torchhk or
• git clone https://github.com/Harry24k/pytorch-custom-utils

from torchhk import *

:rocket: Demos

• RecordManager (code, markdown): RecordManager will help you to keep tracking training records.

• Datasets (code, markdown): Dataset will help you to use torch datasets including split and label-filtering.

  Supported datasets: CIFAR10, CIFAR100, STL10, MNIST, FashionMNIST, SVHN, MNISTM, ImageNet, USPS, TinyImageNet.

• Vis (code, markdown): Vis will help you to visualize torch tensors.

• Transform (code): Transform will help you to change specific layers.

• Trainer (code): Trainer will help you to train a pytorch model.

Contribution

Contribution is always welcome! Use pull requests :blush: