qumphy.callbacks.sleepapnea module

Description: Pytorch lightning callbacks.

class qumphy.callbacks.sleepapnea.SleepApneaLogging(*args: Any, **kwargs: Any)[source]

Bases: BaseLoggingCallback

Pytorch lightning callback for logging for the sleepApnea dataset.

log_epoch_end(trainer, pl_module, stage)[source]

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