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])
176 Dec 07, 2022
268 Dec 24, 2022
520 Jan 04, 2023
11 Apr 12, 2022
24 Dec 06, 2022
11 May 02, 2022
12 Feb 14, 2022
46 Nov 17, 2022
55 Dec 12, 2022
50 Dec 09, 2022
2 Jan 05, 2022
114 Dec 28, 2022
37 Dec 17, 2022
2.6k Jan 01, 2023
196 Jan 04, 2023
12 Jan 04, 2023
12 May 30, 2022
6 Oct 26, 2022
1 May 31, 2022
72 Dec 20, 2022