vis4d.engine.callbacks.ema

Callback for updating EMA model.

Classes

EMACallback([epoch_based, train_connector, ...])

Callback for EMA.

class EMACallback(epoch_based=True, train_connector=None, test_connector=None)[source]

Callback for EMA.

on_train_batch_end(trainer, pl_module, outputs, batch, batch_idx)[source]

Hook to run at the end of a training batch.