https://github.com/borchero/lightkit

Utilities for PyTorch and PyTorch Lightning.
https://github.com/borchero/lightkit

machine-learning pytorch pytorch-lightning

Last synced: 17 days ago
JSON representation

Utilities for PyTorch and PyTorch Lightning.

• Host: GitHub
• URL: https://github.com/borchero/lightkit
• Owner: borchero
• License: mit
• Created: 2021-12-17T15:18:31.000Z (almost 3 years ago)
• Default Branch: main
• Last Pushed: 2024-10-21T20:07:26.000Z (22 days ago)
• Last Synced: 2024-10-22T14:11:44.501Z (22 days ago)
• Topics: machine-learning, pytorch, pytorch-lightning
• Language: Python
• Homepage: https://lightkit.borchero.com
• Size: 164 KB
• Stars: 5
• Watchers: 4
• Forks: 3
• Open Issues: 9
• Metadata Files:
  • Readme: README.md
  • License: LICENSE
  • Codeowners: .github/CODEOWNERS

Awesome Lists containing this project

README

        
# 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](https://lightkit.borchero.com).

## Installation

LightKit is available via `pip`:

```bash

pip install lightkit

```

If you are using [Poetry](https://python-poetry.org/):

```bash

poetry add lightkit

```