Skip to main content

Custom utils for PyTorch

Project description

pytorch-custom-utils

License Pypi

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:

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

torchhk-0.84.1-py3-none-any.whl (18.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page