otx.algorithms.common.tasks.nncf_task#

BaseTask for NNCF.

Classes

NNCFBaseTask()

NNCFBaseTask.

class otx.algorithms.common.tasks.nncf_task.NNCFBaseTask[source]#

Bases: IOptimizationTask

NNCFBaseTask.

static model_builder(config, *args, nncf_model_builder, model_config=None, data_config=None, is_export=False, return_compression_ctrl=False, **kwargs)[source]#: model_builder.

optimize(optimization_type: OptimizationType, dataset: DatasetEntity, output_model: ModelEntity, optimization_parameters: OptimizationParameters | None = None)[source]#: NNCF Optimization.

save_model(output_model: ModelEntity)[source]#: Saving model function for NNCF Task.