Skip to content

Commit fa03d14

Browse files
author
Songki Choi
authored
Merge pull request #1098 from openvinotoolkit/mpa-template-disable
Disable MPA model template
2 parents a1839e2 + 3528567 commit fa03d14

File tree

9 files changed

+5
-5
lines changed

9 files changed

+5
-5
lines changed

external/model-preparation-algorithm/tests/test_ote_api.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def test_reading_classification_cls_incr_model_template(self):
8383
'mobilenet_v3_large_1_cls_incr', 'mobilenet_v3_large_075_cls_incr',
8484
'mobilenet_v3_small_cls_incr']
8585
for model_template in classification_template:
86-
parse_model_template(osp.join('configs', 'classification', model_template, 'template.yaml'))
86+
parse_model_template(osp.join('configs', 'classification', model_template, 'template_experimental.yaml'))
8787

8888
@staticmethod
8989
def generate_label_schema(not_empty_labels, multilabel=False):
@@ -103,7 +103,7 @@ def generate_label_schema(not_empty_labels, multilabel=False):
103103

104104
@staticmethod
105105
def setup_configurable_parameters(template_dir, num_iters=10):
106-
model_template = parse_model_template(osp.join(template_dir, 'template.yaml'))
106+
model_template = parse_model_template(osp.join(template_dir, 'template_experimental.yaml'))
107107
hyper_parameters = create(model_template.hyper_parameters.data)
108108
hyper_parameters.learning_parameters.num_iters = num_iters
109109
return hyper_parameters, model_template
@@ -240,7 +240,7 @@ class MPADetAPI(unittest.TestCase):
240240
def test_reading_detection_cls_incr_model_template(self):
241241
detection_template = ['mobilenetv2_atss_cls_incr', 'resnet50_vfnet_cls_incr']
242242
for model_template in detection_template:
243-
parse_model_template(osp.join('configs', 'detection', model_template, 'template.yaml'))
243+
parse_model_template(osp.join('configs', 'detection', model_template, 'template_experimental.yaml'))
244244

245245
def init_environment(
246246
self,
@@ -455,7 +455,7 @@ class MPASegAPI(unittest.TestCase):
455455
def test_reading_segmentation_cls_incr_model_template(self):
456456
segmentation_template = ['ocr-lite-hrnet-18-cls-incr']
457457
for model_template in segmentation_template:
458-
parse_model_template(osp.join('configs', 'segmentation', model_template, 'template.yaml'))
458+
parse_model_template(osp.join('configs', 'segmentation', model_template, 'template_experimental.yaml'))
459459

460460
@staticmethod
461461
def generate_label_schema(label_names):
@@ -537,7 +537,7 @@ def init_environment(self, params, model_template, number_of_images=10):
537537

538538
@staticmethod
539539
def setup_configurable_parameters(template_dir, num_iters=10):
540-
model_template = parse_model_template(osp.join(template_dir, 'template.yaml'))
540+
model_template = parse_model_template(osp.join(template_dir, 'template_experimental.yaml'))
541541

542542
hyper_parameters = create(model_template.hyper_parameters.data)
543543
hyper_parameters.learning_parameters.learning_rate_fixed_iters = 0

0 commit comments

Comments
 (0)