neural_compressor.model.mxnet_model
¶
Class for MXNet model.
Module Contents¶
Classes¶
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.