vis4d.engine.callbacks.logging

This module contains utilities for callbacks.

Classes

LoggingCallback(*args[, refresh_rate])

Callback for logging.

class LoggingCallback(*args, refresh_rate=50, **kwargs)

Callback for logging.

__init__(*args, refresh_rate=50, **kwargs)

Init callback.

on_train_epoch_start(trainer, pl_module)

Hook to run at the start of a training epoch.

Return type:

None

on_train_batch_start(trainer, pl_module, batch, batch_idx)

Hook to run at the start of a training batch.

Return type:

None

on_train_batch_end(trainer, pl_module, outputs, batch, batch_idx)

Hook to run at the end of a training batch.

Return type:

None

on_validation_epoch_start(trainer, pl_module)

Hook to run at the start of a validation epoch.

Return type:

None

on_validation_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)

Wait for on_validation_batch_end PL hook to call ‘process’.

Return type:

None

on_test_epoch_start(trainer, pl_module)

Hook to run at the start of a testing epoch.

Return type:

None

on_test_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)

Hook to run at the end of a testing batch.

Return type:

None