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 callback.

on_test_batch_end(trainer_state, model, outputs, batch, batch_idx, dataloader_idx=0)

Hook to run at the end of a testing batch.

Return type:

None

on_test_epoch_start(trainer_state, model)

Hook to run at the start of a testing epoch.

Return type:

None

on_train_batch_end(trainer_state, model, loss_module, outputs, batch, batch_idx)

Hook to run at the end of a training batch.

Return type:

Optional[Dict[str, Union[float, int, Tensor]]]

on_train_batch_start(trainer_state, model, loss_module, batch, batch_idx)

Hook to run at the start of a training batch.

Return type:

None

on_train_epoch_start(trainer_state, model, loss_module)

Hook to run at the start of a training epoch.

Return type:

None