lightning_trainer

This is summary documentation for hippynn.experiment.lightning_trainer module. Click here for full documentation.

Links to Full Documentation:

Pytorch Lightning training interface.

This module is somewhat experimental. Using pytorch lightning successfully in a distributed context may require understanding and adjusting the various settings related to parallelism, e.g. multiprocessing context, torch ddp backend, and how they interact with your HPC environment.

Some features of hippynn experiments may not be implemented yet.

  • The plotmaker is currently not supported.

Classes

HippynnDataModule(*args, **kwargs)

HippynnLightningModule(*args, **kwargs)

A pytorch lightning module for running a hippynn experiment.

LightingPrintStagesCallback(*args, **kwargs)

This callback is for debugging only.