dmx.compressor.modeling.model.Model

class dmx.compressor.modeling.model.Model(body, head=Identity(), tail=Identity(), hf: bool = False, concrete_args: Dict[str, Any] | None = None, **kwargs)

This is a backward-compatible placeholder for legacy models. It is not recommended to use this container; instead, use DmxSimplePipeline directly. TODO: to be deprecated

__init__(body, head=Identity(), tail=Identity(), hf: bool = False, concrete_args: Dict[str, Any] | None = None, **kwargs) → None

Methods

__init__(body[, head, tail, hf, concrete_args])
configure(dmx_config_dict[, ...])
counting_flops([zero])
eval()
get_model_by_name(model_name)
named_dmx_models()	Returns a generator of named DmxModel instances
named_dmx_modules()
to([torch_device])
transform(dmx_config_dict[, ...])

Attributes

dmx_config_dict
op_set	Returns a set of unique ops present in the model