Utilities for PyTorch and PyTorch Lightning.
Project description
LightKit
LightKit provides simple utilities for working with PyTorch and PyTorch Lightning. At the moment, it provides three simple features:
- A data loader for tabular data that is orders of magnitude faster than PyTorch's builtin data loader for medium-sized datasets and larger ones.
- A mixin for modules that allows to save not only weights, but also the configuration to the file system such that it is easier to retrieve trained models.
- A typed base class for estimators that enables users to easily create estimators with PyTorch and PyTorch Lightning which are fully compatible with Scikit-learn.
For more details, consult the documentation.
Installation
LightKit is available via pip
:
pip install lightkit
If you are using Poetry:
poetry add lightkit
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
lightkit-0.5.0.tar.gz
(13.0 kB
view hashes)
Built Distribution
lightkit-0.5.0-py3-none-any.whl
(17.2 kB
view hashes)