qumphy.callbacks.deepbeat module

File: qumphy/models/lightning_callbacks.py Project: 22HLT01 QUMPHY Contact: oskar.pfeffer@ptb.de Gitlab: https://gitlab.com/qumphy Description: Pytorch lightning callbacks.

class qumphy.callbacks.deepbeat.DeepBeatLogging(*args: Any, **kwargs: Any)[source]

Bases: BaseLoggingCallback

Pytorch lightning callback for logging for the DeepBeat dataset.

log_epoch_end(trainer, pl_module, stage)[source]

Override this method to log the metrics at the end of the epoch.

class qumphy.callbacks.deepbeat.DeepBeatLogging_KGLoss(*args: Any, **kwargs: Any)[source]

Bases: DeepBeatLogging

on_test_epoch_end(trainer, pl_module)[source]
class qumphy.callbacks.deepbeat.DeepBeatLogging_MCD(*args: Any, **kwargs: Any)[source]

Bases: DeepBeatLogging

Pytorch lightning callback for logging for the DeepBeat dataset.

Initialize with a list of quantities to log.

Available quantities:

  • loss

  • qumphy

on_test_epoch_end(trainer, pl_module)[source]