neural_compressor.model.base_model

Base model for multiple framework backends.

Module Contents

Classes

BaseModel

Base class of all neural_compressor.model, will play graph role.

class neural_compressor.model.base_model.BaseModel(model, **kwargs)

Base class of all neural_compressor.model, will play graph role.

abstract property model

Return model itself.

abstract property graph_info

Node_type’.

Type:

Return a dict with content ‘Node

abstract save(root, *args, **kwargs)

Abstract method of model saving.

abstract export(save_path: str, conf)

Abstract method of model convertion to ONNX.

abstract framework()

Abstract method of model framework.