Skip to content

Commit a6c5f4d

Browse files
author
Songki Choi
authored
Fix optimization feature backward compatibility issue (#2141)
Temporary disable quantization for 2 models due to Geti project upgrade issue Signed-off-by: Songki Choi <[email protected]>
1 parent 8fae2bb commit a6c5f4d

File tree

3 files changed

+8
-3
lines changed

3 files changed

+8
-3
lines changed

otx/algorithms/common/tasks/nncf_task.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,12 @@ def _set_attributes_by_hyperparams(self):
112112
self._optimization_methods = [OptimizationMethod.FILTER_PRUNING]
113113
self._precision = [ModelPrecision.FP32]
114114
return
115-
raise RuntimeError("Not selected optimization algorithm")
115+
# FIXEME: Error rasing should be re-enabled after Geti issue resolved
116+
# raise RuntimeError("Not selected optimization algorithm")
117+
logger.warning("Not selected optimization algorithm. Defaults to quantization")
118+
self._nncf_preset = "nncf_quantization"
119+
self._optimization_methods = [OptimizationMethod.QUANTIZATION]
120+
self._precision = [ModelPrecision.INT8]
116121

117122
def _init_train_data_cfg(self, dataset: DatasetEntity):
118123
logger.info("init data cfg.")

otx/algorithms/detection/configs/instance_segmentation/efficientnetb2b_maskrcnn/template.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ hyper_parameters:
4040
default_value: 1
4141
nncf_optimization:
4242
enable_quantization:
43-
default_value: true
43+
default_value: false
4444
enable_pruning:
4545
default_value: false
4646
pruning_supported:

otx/algorithms/segmentation/configs/ocr_lite_hrnet_x_mod3/template.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ hyper_parameters:
3939
default_value: 300
4040
nncf_optimization:
4141
enable_quantization:
42-
default_value: true
42+
default_value: false
4343
enable_pruning:
4444
default_value: false
4545
pruning_supported:

0 commit comments

Comments
 (0)