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.13.11.50.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.13.11.50.tar.gz.

File metadata

File hashes

Hashes for lightning_trainer_utils-2025.5.13.11.50.tar.gz
Algorithm Hash digest
SHA256 af21e263269acc13eb39f75b21f3cff5c3e8a677998bfdc19d0f796a3ffe6d6d
MD5 fb85aa6546b9424de3f21cf905579c84
BLAKE2b-256 941a13fceec4a782b86374fa03b4dea9c32c6240b91287a8b7d70b23e837150e

See more details on using hashes here.

Provenance

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

File metadata

File hashes

Hashes for lightning_trainer_utils-2025.5.13.11.50-py3-none-any.whl
Algorithm Hash digest
SHA256 8fff88ca4565887c70de88500d4c778d934256fc75c192e3694e3bbc620b415b
MD5 7ff460e42e49fa26034f5e12f0371670
BLAKE2b-256 18060b4636dd0461bbfe00c5f91c0fe4dc2cc4ac43ec0618a5de0a0a96d3c337

See more details on using hashes here.

Provenance

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