Core APIs for deep learning.
Project description
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
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
Built Distribution
Hashes for core_pytorch_utils-1.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4f9daafed1cd22c149d18eb00691cc6c09fdb2761c96d5c4678d3cb197af3ce1 |
|
MD5 | 640e8190c684f5f0eea4983586120178 |
|
BLAKE2b-256 | b27ac1fcf7da0d43dfaebc602fd2fa765027a3614c8c2d2e46a0132ce48ecf79 |