vis4d.engine.callbacks.visualizer¶

This module contains utilities for callbacks.

Classes

VisualizerCallback(*args, visualizer[, ...])

Callback for model visualization.

class VisualizerCallback(*args, visualizer, visualize_train=False, show=False, save_to_disk=True, save_prefix=None, **kwargs)[source]¶

Callback for model visualization.

Init callback.

Parameters:

visualizer (Visualizer) – Visualizer.
visualize_train (bool) – If the training data should be visualized. Defaults to False.
save_prefix (str) – Output directory for saving the visualizations.
show (bool) – If the visualizations should be shown. Defaults to False.
save_to_disk (bool) – If the visualizations should be saved to disk. Defaults to True.

on_test_batch_end(trainer_state, model, outputs, batch, batch_idx, dataloader_idx=0)[source]¶

Hook to run at the end of a testing batch.

on_train_batch_end(trainer_state, model, loss_module, outputs, batch, batch_idx)[source]¶

Hook to run at the end of a training batch.

Setup callback.