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-2025.5.14.15.30.tar.gz (5.7 kB view details)

Uploaded Source

Built Distribution

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

File details

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

File metadata

File hashes

Hashes for lightning_trainer_utils-2025.5.14.15.30.tar.gz
Algorithm Hash digest
SHA256 6f5776539738701c6444b44c52745f88a1506b5b9363bfe39dcb3bafa9f937c7
MD5 645432e43cb83f671cba1e3831474f88
BLAKE2b-256 fd8a109818ffa234c3ea099f5ec54fb82d3c0844bec262e781db435ca83965d9

See more details on using hashes here.

Provenance

The following attestation bundles were made for lightning_trainer_utils-2025.5.14.15.30.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-2025.5.14.15.30-py3-none-any.whl.

File metadata

File hashes

Hashes for lightning_trainer_utils-2025.5.14.15.30-py3-none-any.whl
Algorithm Hash digest
SHA256 1ec3326efcbd5b9df7bec0a765f547d590d6ca82a3463ae54705f5c66f81dfc9
MD5 77c28adb8a67ee2dabd8c1f92d2bf969
BLAKE2b-256 683625b7ce4e9057785f843be810a9cffb94cb96f1428a24ba236d2ea7769aed

See more details on using hashes here.

Provenance

The following attestation bundles were made for lightning_trainer_utils-2025.5.14.15.30-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