pyutils
A python/pytorch utility library
News
- v0.0.1 available. Feedbacks are highly welcomed!
Installation
git clone https://github.com/JeremieMelo/pyutility.git
cd pyutility
pip3 install --editable .
Features
- Support pytorch training utility and datasets.
TODOs
- Support lr_scheduler
- Support trainer
Dependencies
- Python >= 3.6
- PyTorch >= 1.8.0
- Tensorflow >= 2.5.0
- Others are listed in requirements.txt
Files
| 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 |
Contact
Jiaqi Gu ([email protected])
647 Jan 05, 2023
85 Dec 15, 2022
51 Nov 29, 2022
29 Nov 28, 2022
1 Feb 13, 2022
4 Dec 14, 2022
1 Dec 10, 2021
46 Nov 20, 2022
118 Dec 21, 2022
649 Dec 27, 2022
4 Dec 28, 2021
385 Dec 28, 2022
114 Dec 30, 2022
2k Dec 31, 2022
50 Dec 28, 2022
159 Dec 30, 2022
2.4k Dec 25, 2022
934 Jan 08, 2023
5 Sep 21, 2022
104 Dec 15, 2022