neural_compressor.model.nets_factory

TF-Slim nets factory.

Module Contents

Classes

TFSlimNetsFactory

TF-Slim nets factory.
class neural_compressor.model.nets_factory.TFSlimNetsFactory

Bases: object

TF-Slim nets factory.

register(name, model_func, input_shape, arg_scope, **kwargs)

Register a model to TFSlimNetsFactory.

Parameters:

  • name (str) – name of a model.

  • model_func (_type_) – model that built from slim.

  • input_shape (_type_) – input tensor shape.

  • arg_scope (_type_) – slim arg scope that needed.