dmx.compressor.modeling.model.DmxSimplePipeline

class dmx.compressor.modeling.model.DmxSimplePipeline(model_dict: OrderedDict, preproc=Identity(), postproc=Identity(), hf: bool = False, concrete_args_dict: Dict | None = None)

__init__(model_dict: OrderedDict, preproc=Identity(), postproc=Identity(), hf: bool = False, concrete_args_dict: Dict | None = None) → None

Methods

__init__(model_dict[, preproc, postproc, ...])
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