├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── bld.bat ├── meta.yaml ├── pyproject.toml ├── pyutils ├── __init__.py ├── activation.py ├── compute.py ├── config.py ├── datasets │ ├── __init__.py │ ├── builder.py │ ├── cars.py │ ├── cifar10.py │ ├── cifar100.py │ ├── dogs.py │ ├── fashionmnist.py │ ├── flowers.py │ ├── mnist.py │ ├── svhn.py │ ├── tinyimagenet.py │ └── vowel.py ├── distribution_sampler.py ├── general.py ├── initializer.py ├── loss │ ├── __init__.py │ ├── alpha_divergence.py │ ├── ce_smooth.py │ ├── dkd.py │ ├── dual_focal.py │ ├── focal_smooth.py │ ├── kd.py │ └── utils.py ├── lr_scheduler │ ├── __init__.py │ └── warmup_cosine_restart.py ├── metric.py ├── normalization.py ├── optimizer │ ├── __init__.py │ ├── adam_gc.py │ ├── dadapt_adam.py │ ├── dadapt_sgd.py │ ├── gld.py │ ├── lookahead.py │ ├── nesterov.py │ ├── prodigy.py │ ├── radam.py │ ├── sam.py │ ├── sgd_gc.py │ └── smtp.py ├── plot.py ├── quant │ ├── __init__.py │ └── lsq.py ├── quantize.py ├── torch_train.py ├── typing.py └── version.py ├── setup.sh └── unitest ├── compute_test.py ├── dataset_test.py ├── general_test.py ├── loss_test.py ├── quantize_test.py └── test.yaml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/README.md -------------------------------------------------------------------------------- /bld.bat: -------------------------------------------------------------------------------- 1 | "%PYTHON%" -m pip install --editable . 2 | if errorlevel 1 exit 1 3 | -------------------------------------------------------------------------------- /meta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/meta.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pyutils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/__init__.py -------------------------------------------------------------------------------- /pyutils/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/activation.py -------------------------------------------------------------------------------- /pyutils/compute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/compute.py -------------------------------------------------------------------------------- /pyutils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/config.py -------------------------------------------------------------------------------- /pyutils/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/__init__.py -------------------------------------------------------------------------------- /pyutils/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/builder.py -------------------------------------------------------------------------------- /pyutils/datasets/cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/cars.py -------------------------------------------------------------------------------- /pyutils/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/cifar10.py -------------------------------------------------------------------------------- /pyutils/datasets/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/cifar100.py -------------------------------------------------------------------------------- /pyutils/datasets/dogs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/dogs.py -------------------------------------------------------------------------------- /pyutils/datasets/fashionmnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/fashionmnist.py -------------------------------------------------------------------------------- /pyutils/datasets/flowers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/flowers.py -------------------------------------------------------------------------------- /pyutils/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/mnist.py -------------------------------------------------------------------------------- /pyutils/datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/svhn.py -------------------------------------------------------------------------------- /pyutils/datasets/tinyimagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/tinyimagenet.py -------------------------------------------------------------------------------- /pyutils/datasets/vowel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/datasets/vowel.py -------------------------------------------------------------------------------- /pyutils/distribution_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/distribution_sampler.py -------------------------------------------------------------------------------- /pyutils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/general.py -------------------------------------------------------------------------------- /pyutils/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/initializer.py -------------------------------------------------------------------------------- /pyutils/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/__init__.py -------------------------------------------------------------------------------- /pyutils/loss/alpha_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/alpha_divergence.py -------------------------------------------------------------------------------- /pyutils/loss/ce_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/ce_smooth.py -------------------------------------------------------------------------------- /pyutils/loss/dkd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/dkd.py -------------------------------------------------------------------------------- /pyutils/loss/dual_focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/dual_focal.py -------------------------------------------------------------------------------- /pyutils/loss/focal_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/focal_smooth.py -------------------------------------------------------------------------------- /pyutils/loss/kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/kd.py -------------------------------------------------------------------------------- /pyutils/loss/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/loss/utils.py -------------------------------------------------------------------------------- /pyutils/lr_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/lr_scheduler/__init__.py -------------------------------------------------------------------------------- /pyutils/lr_scheduler/warmup_cosine_restart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/lr_scheduler/warmup_cosine_restart.py -------------------------------------------------------------------------------- /pyutils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/metric.py -------------------------------------------------------------------------------- /pyutils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/normalization.py -------------------------------------------------------------------------------- /pyutils/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/__init__.py -------------------------------------------------------------------------------- /pyutils/optimizer/adam_gc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/adam_gc.py -------------------------------------------------------------------------------- /pyutils/optimizer/dadapt_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/dadapt_adam.py -------------------------------------------------------------------------------- /pyutils/optimizer/dadapt_sgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/dadapt_sgd.py -------------------------------------------------------------------------------- /pyutils/optimizer/gld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/gld.py -------------------------------------------------------------------------------- /pyutils/optimizer/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/lookahead.py -------------------------------------------------------------------------------- /pyutils/optimizer/nesterov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/nesterov.py -------------------------------------------------------------------------------- /pyutils/optimizer/prodigy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/prodigy.py -------------------------------------------------------------------------------- /pyutils/optimizer/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/radam.py -------------------------------------------------------------------------------- /pyutils/optimizer/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/sam.py -------------------------------------------------------------------------------- /pyutils/optimizer/sgd_gc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/sgd_gc.py -------------------------------------------------------------------------------- /pyutils/optimizer/smtp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/optimizer/smtp.py -------------------------------------------------------------------------------- /pyutils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/plot.py -------------------------------------------------------------------------------- /pyutils/quant/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/quant/__init__.py -------------------------------------------------------------------------------- /pyutils/quant/lsq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/quant/lsq.py -------------------------------------------------------------------------------- /pyutils/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/quantize.py -------------------------------------------------------------------------------- /pyutils/torch_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/torch_train.py -------------------------------------------------------------------------------- /pyutils/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/typing.py -------------------------------------------------------------------------------- /pyutils/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/pyutils/version.py -------------------------------------------------------------------------------- /setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/setup.sh -------------------------------------------------------------------------------- /unitest/compute_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/unitest/compute_test.py -------------------------------------------------------------------------------- /unitest/dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/unitest/dataset_test.py -------------------------------------------------------------------------------- /unitest/general_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/unitest/general_test.py -------------------------------------------------------------------------------- /unitest/loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/unitest/loss_test.py -------------------------------------------------------------------------------- /unitest/quantize_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeremieMelo/pyutility/HEAD/unitest/quantize_test.py -------------------------------------------------------------------------------- /unitest/test.yaml: -------------------------------------------------------------------------------- 1 | test: 1 2 | --------------------------------------------------------------------------------