Skip to main content

A library of classes and functions for working with PyTorch.

Project description

firekit

Firekit is a library of classes and functions for training and evaluating PyTorch models. The main focus of the library is a Trainer class that performs the standard training and evaluation loops, reports the training and evaluation loss and the evaluation performance on user-defined metrics, saves the model state when performance improves, and reloads the best model at the end of training.

This project exists to support my work. It is in active development and the API is not stable.

Installation

Install with pip or pipenv in the normal way.

pip install firekit

Use the --extra-index-url argument to install PyTorch for CUDA as a dependency. For example, use the following to get PyTorch with CUDA 11.3.

pip install firekit --extra-index-url https://download.pytorch.org/whl/cu113

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

firekit-0.0.8.tar.gz (5.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

firekit-0.0.8-py3-none-any.whl (9.4 kB view details)

Uploaded Python 3

File details

Details for the file firekit-0.0.8.tar.gz.

File metadata

  • Download URL: firekit-0.0.8.tar.gz
  • Upload date:
  • Size: 5.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.1 CPython/3.10.6

File hashes

Hashes for firekit-0.0.8.tar.gz
Algorithm Hash digest
SHA256 ab528d3aca34aabec7d7fb988847430435b82171d04d723247a9aa5317e435f4
MD5 9490e5b24aecf97c2f7b9c57cbd01d8e
BLAKE2b-256 9207b67a38ad8f2741a75c0b27818a6cab8b46217013ad95fb47400f1528caf3

See more details on using hashes here.

File details

Details for the file firekit-0.0.8-py3-none-any.whl.

File metadata

  • Download URL: firekit-0.0.8-py3-none-any.whl
  • Upload date:
  • Size: 9.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.1 CPython/3.10.6

File hashes

Hashes for firekit-0.0.8-py3-none-any.whl
Algorithm Hash digest
SHA256 698842b0c6a4b253017f29c2ae2609fabb79f7cfcf1be1c6915f2205ef03afb2
MD5 10814c291dead64224195ff66efa6db5
BLAKE2b-256 24d30f72ca75352c7428ddab2a2ec79a3e98152a8b9e4c637a11432c4dd026b6

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page