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])
117 Nov 21, 2022
169 Dec 20, 2022
25 Oct 31, 2022
5 Jul 18, 2022
236 Jan 01, 2023
7 Oct 11, 2022
58 Oct 11, 2022
2 Oct 10, 2022
688 Jan 06, 2023
50 Dec 03, 2022
214 Dec 28, 2022
100 Nov 15, 2022
1.3k Jan 03, 2023
4 May 28, 2022
5.2k Dec 28, 2022
132 Jan 02, 2023
1 Nov 22, 2021
357 Jan 04, 2023
18 Jul 19, 2022
7 Feb 15, 2022