vis4d.model.adapter.flops

Adapter for counting flops in a model.

Classes

FlopsModelAdapter(model, data_connector)

Adapter for the model to count flops.

class FlopsModelAdapter(model, data_connector)[source]

Adapter for the model to count flops.

__init__(model, data_connector)[source]: Initialize the adapter.

forward(*args)[source]

Forward pass through the model.

Return type:: Any