pyutils
A python/pytorch utility library
News
- v0.0.3.2 available.
- v0.0.2 available. Added new datasets and quantization!
- v0.0.1 available. Feedbacks are highly welcomed!
Installation
conda install scopex/label/ScopeX::torchonn-pyutils
pip install torchonn-pyutils --no-build-isolation
or install from cloned codes from github if you would like to modify the code (recommend)
git clone https://github.com/JeremieMelo/pyutility.git
cd pyutility
./setup.sh
To remove the package:
pip uninstall torchonn_pyutils
Usage
import pyutils
Features
- Support pytorch training utility and datasets.
TODOs
- Support lr_scheduler
- Support trainer
Dependencies
- Python >= 3.10
- PyTorch >= 2.0
- Tensorflow >= 2.5.0
- Others are listed in requirements.txt
Files
| File | Description |
|---|---|
| datasets/ | Defines different datasets and builder |
| loss/ | Defines different loss functions/criterions |
| optimizer/ | Defines different optimizers |
| lr_scheduler/ | Defines different learning rate schedulers |
| quant/ | Defines different weight/activation quantizers |
| activation.py | Activation functions |
| compute.py | functions related to computing |
| config.py | Hierarchical yaml configuration file parser |
| distribution_sampler.py | Sample from customized distributions |
| general.py | Common helper functions |
| initializer.py | Initialization methods for PyTorch Parameters |
| loss.py | Loss functions for PyTorch model training |
| quantize.py | Quantization functions |
| torch_train.py | Helper functions for torch training |
| typing.py | Defines common types |
Contact
Jiaqi Gu (jqgu@utexas.edu)
1 maintainer
Metadata
1 maintainer
Release files for torchonn-pyutils 0.0.3.3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| torchonn_pyutils-0.0.3.3.tar.gz | 76.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| torchonn_pyutils-0.0.3.3-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 180.3 kB
Release files / torchonn_pyutils-0.0.3.3.tar.gz
| Download URL | torchonn_pyutils-0.0.3.3.tar.gz |
|---|---|
| Size | 76.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
596e4d746a61dea9174999e693b6b5d42cc5bc4ffcac28a019e3fcb89ea3cdaf
|
|
BLAKE2b-256 checksum How to use checksums |
a2c7c5a0935fa4e37a069a42ec7ce3fde773f234eec1ae5beb6bda8739693426
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.11.10
|
Release files / torchonn_pyutils-0.0.3.3-py3-none-any.whl
| Download URL | torchonn_pyutils-0.0.3.3-py3-none-any.whl |
|---|---|
| Size | 104.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
aab208c146ed61051093fbef782a566cfbfbd5db697f00e88d4ea918ca9f85a7
|
|
BLAKE2b-256 checksum How to use checksums |
99f31e9ff703d1a90fa77f00d12f82f637eee6f2c90ad285d2992adc23a10197
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.11.10
|
