vis4d.model.adapter

Model adapters.

class ModelEMAAdapter(model, decay=0.9998, use_ema_during_test=True, device=None)

Torch module with Exponential Moving Average (EMA).

Parameters:

• model (nn.Module) – Model to apply EMA.

• decay (float) – Decay factor for EMA. Defaults to 0.9998.

• use_ema_during_test (bool) – Use EMA model during testing. Defaults to True.

• device (torch.device | None) – Device to use. Defaults to None.

Init ModelEMAAdapter class.

forward(*args, **kwargs)

Forward pass with original model.

Return type:

Any

set(model)

Copy model params into the internal EMA.

Return type:

None

update(steps)

Update the internal EMA model.

Return type:

None

class ModelExpEMAAdapter(model, decay=0.9998, warmup_steps=2000, use_ema_during_test=True, device=None)

Exponential Moving Average (EMA) with exponential decay strategy.

Used by YOLOX.

Parameters:

• model (nn.Module) – Model to apply EMA.

• decay (float) – Decay factor for EMA. Defaults to 0.9998.

• warmup_steps (int) – Number of warmup steps for decay. Use a smaller decay early in training and gradually anneal to the set decay value to update the EMA model smoothly.

• use_ema_during_test (bool) – Use EMA model during testing. Defaults to True.

• device (torch.device | None) – Device to use. Defaults to None.

Init ModelEMAAdapter class.

update(steps)

Update the internal EMA model.

Return type:

None

Modules

vis4d.model.adapter.ema	Exponential Moving Average (EMA) for PyTorch models.
vis4d.model.adapter.flops	Adapter for counting flops in a model.