combustion.lightning

Utilities to facilitate operation with PyTorch Lightning

General Callbacks

Visualization Callbacks

Combustion provides a set of callbacks that integrate with PyTorch Lightning to provide a simplistic yet powerful visualization feature set. Tensors to be visualized are attached as model attributes during training or inference steps, and callbacks read these attributes to create visualizations.

Metrics