Skip to main content

Core APIs for deep learning.

Project description

drawing

Docs GithubAction Codecov License

Core PyTorch Utils (CPU)

This package is a light-weight core library that provides the most common and essential functionalities shared in various deep learning tasks:

  • Trainer: does tedious training logic for you.
  • LRWarmupScheduler: wraps all standard PyTorch LR scheduler to support warmup.
  • ConfigArgumentParser: provides an argument parser that supports loading a YAML configuration file.
  • ......

You can find a brief Chinese introduction at zhihu.

Installation

From PyPI.

pip install core-pytorch-utils

Or from source.

git clone https://github.com/serend1p1ty/core-pytorch-utils.git
cd core-pytorch-utils
pip install -r requirements.txt
pip install -v -e .

Getting Started

In examples/ folder, we show how to use our Trainer to train a CNN on MINIST.

It is strongly recommended that you run this code before using the CPU library.

Advanced

Learn more from our documentaion.

Contributing

Pull request is welcomed! Before submitting a PR, DO NOT forget to run ./dev/linter.sh that provides syntax checking and code style optimation.

License

CPU is released under the MIT License.

Acknowledgments

We refered mmcv, detectron2 and pytorch-image-models when develping CPU.

Project details


Download files

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

Source Distribution

core-pytorch-utils-1.0.4.tar.gz (28.6 kB view hashes)

Uploaded Source

Built Distribution

core_pytorch_utils-1.0.4-py3-none-any.whl (35.0 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