Detector
pnpxai.core.detector.detector
DEFAULT_MODULE_TYPES_TO_DETECT = (Linear, Convolution, RNN, LSTM, Attention, Embedding)
module-attribute
Tracer
Bases: Tracer
is_leaf_module(m: nn.Module, module_qualified_name: str) -> bool
symbolic_trace(model: nn.Module) -> fx.GraphModule
extract_graph_data(graph_module: fx.GraphModule)
detect_model_architecture(model: Model, targets: Optional[Tuple[ModuleType]] = None) -> Set[ModuleType]
A function detecting architecture for a given model.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
model
|
Model
|
The machine learning model to be detected |
required |
Returns:
| Name | Type | Description |
|---|---|---|
ModelArchitectureSummary |
Set[ModuleType]
|
A summary of model architecture |