neural_compressor.model.mxnet_model

Class for MXNet model.

Module Contents

Classes

MXNetModel

Build MXNet model.

class neural_compressor.model.mxnet_model.MXNetModel(model, **kwargs)

Bases: neural_compressor.model.base_model.BaseModel

Build MXNet model.

property model

Return model itself.

framework()

Return framework.

save(root=None)

Save MXNet model.