Skip to main content

A Python package for using PyTorch Lightning with custom callbacks and model wrappers.

Project description

PyTorch Lightning Trainer Utilities

ML Model Assumptions

forward

  • The batch will be passed as kwargs to forward function of the ML model.
self.model(**batch)

return

  • ML model should return a dict with the following keys:
    • loss
    • loss-dict
    • output [optional]

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

lightning_trainer_utils-0.1.0.tar.gz (5.3 kB view details)

Uploaded Source

Built Distribution

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

lightning_trainer_utils-0.1.0-py3-none-any.whl (6.3 kB view details)

Uploaded Python 3

File details

Details for the file lightning_trainer_utils-0.1.0.tar.gz.

File metadata

  • Download URL: lightning_trainer_utils-0.1.0.tar.gz
  • Upload date:
  • Size: 5.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.12.8

File hashes

Hashes for lightning_trainer_utils-0.1.0.tar.gz
Algorithm Hash digest
SHA256 aff49030e5fd3330ea760b126ed7642424475c60b3e44c9af1d92aa466fb55c5
MD5 32736896858746d4a535f98f05404c3a
BLAKE2b-256 290ff9d00912a715340f9ad2ea82679f5e7a29b203d15280eabf4885746c92eb

See more details on using hashes here.

Provenance

The following attestation bundles were made for lightning_trainer_utils-0.1.0.tar.gz:

Publisher: workflow.yaml on manavmahan/lightning-trainer-utils

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file lightning_trainer_utils-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for lightning_trainer_utils-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 f8ba18572f78dd32b1a2b12c7d71e2ad42d686454c42a010dcb2d6a4b11e83be
MD5 e252b9015e08c3cabcfc297482e65baf
BLAKE2b-256 e6c3e3ad50762c2dc3c869373e978fba5965360d6616fe58c01cedf280723cee

See more details on using hashes here.

Provenance

The following attestation bundles were made for lightning_trainer_utils-0.1.0-py3-none-any.whl:

Publisher: workflow.yaml on manavmahan/lightning-trainer-utils

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

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