combustion.lightning¶
Utilities to facilitate operation with PyTorch Lightning
combustion.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.