A python/pytorch utility library
- v0.0.2 available. Added new datasets and quantization!
- v0.0.1 available. Feedbacks are highly welcomed!
git clone https://github.com/JeremieMelo/pyutility.git
cd pyutility
pip3 install --editable .- Support pytorch training utility and datasets.
- Support lr_scheduler
- Support trainer
- Python >= 3.6
- PyTorch >= 1.8.0
- Tensorflow >= 2.5.0
- Others are listed in requirements.txt
| File | Description |
|---|---|
| datasets/ | Defines different datasets and builder |
| optimizer/ | Defines different optimizers |
| 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 |
Jiaqi Gu (jqgu@utexas.edu)