File tree Expand file tree Collapse file tree 3 files changed +8
-3
lines changed
detection/configs/instance_segmentation/efficientnetb2b_maskrcnn
segmentation/configs/ocr_lite_hrnet_x_mod3 Expand file tree Collapse file tree 3 files changed +8
-3
lines changed Original file line number Diff line number Diff 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." )
Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments