@@ -83,7 +83,7 @@ def test_reading_classification_cls_incr_model_template(self):
83
83
'mobilenet_v3_large_1_cls_incr' , 'mobilenet_v3_large_075_cls_incr' ,
84
84
'mobilenet_v3_small_cls_incr' ]
85
85
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' ))
87
87
88
88
@staticmethod
89
89
def generate_label_schema (not_empty_labels , multilabel = False ):
@@ -103,7 +103,7 @@ def generate_label_schema(not_empty_labels, multilabel=False):
103
103
104
104
@staticmethod
105
105
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' ))
107
107
hyper_parameters = create (model_template .hyper_parameters .data )
108
108
hyper_parameters .learning_parameters .num_iters = num_iters
109
109
return hyper_parameters , model_template
@@ -240,7 +240,7 @@ class MPADetAPI(unittest.TestCase):
240
240
def test_reading_detection_cls_incr_model_template (self ):
241
241
detection_template = ['mobilenetv2_atss_cls_incr' , 'resnet50_vfnet_cls_incr' ]
242
242
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' ))
244
244
245
245
def init_environment (
246
246
self ,
@@ -455,7 +455,7 @@ class MPASegAPI(unittest.TestCase):
455
455
def test_reading_segmentation_cls_incr_model_template (self ):
456
456
segmentation_template = ['ocr-lite-hrnet-18-cls-incr' ]
457
457
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' ))
459
459
460
460
@staticmethod
461
461
def generate_label_schema (label_names ):
@@ -537,7 +537,7 @@ def init_environment(self, params, model_template, number_of_images=10):
537
537
538
538
@staticmethod
539
539
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' ))
541
541
542
542
hyper_parameters = create (model_template .hyper_parameters .data )
543
543
hyper_parameters .learning_parameters .learning_rate_fixed_iters = 0
0 commit comments