Framework Integrations¶

FastAI¶

class labml.utils.fastai.LabMLFastAICallback(**kwargs)[source]¶: FastAI callback integration. Pass an instance of this class to FastAI learner as argument cbs. FastAI will call relavent mehtods of this class to log metrics.

Keras¶

class labml.utils.keras.LabMLKerasCallback(save_batch_frequency: int = 1)[source]¶: Keras callback integration. Pass an instance of this class to Keras model fit method as argument callbacks. Keras will call relavent mehtods of this class to log metrics.

PyTorch Lightning¶

PyTorch Lightening Sample

class labml.utils.lightning.LabMLLightningLogger[source]¶: PyTorch Lightening logger integration. Pass an instance of this class to pytorch_lightning.Training as argument logger. PyTorch Lightening will call relavent mehtods of this class to log hyper-parameters and metrics.