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.

Initialize the adapter.

forward(*args)[source]¶

Forward pass through the model.

Return type:: Any