`causica.lightning.loggers`¶

Module Contents¶

`BufferingMlFlowLogger`	MlFlowLogger that buffers metrics on logging and flushes on finalize or when the buffer is full.

class causica.lightning.loggers.BufferingMlFlowLogger(buffer_size: int, *args, **kwargs)[source]¶

Bases: pytorch_lightning.loggers.MLFlowLogger

MlFlowLogger that buffers metrics on logging and flushes on finalize or when the buffer is full.

_buffer_log_batch_metrics(original_log_batch)[source]¶: Returns a decorated log_batch that buffers metrics and flushes them when the buffer is full.

get_buffer_count() → int[source]¶: Return the current number of buffered messages.

finalize(*args, **kwargs) → None[source]¶

Do any processing that is necessary to finalize an experiment.

Parameters:¶

status: Status that the experiment finished with (e.g. success, failed, aborted)