Skip to content

Commit 5ea7a47

Browse files
committed
add fp16 back for efficientnetb2b_maskrcnn and add more tests
1 parent d975d18 commit 5ea7a47

File tree

2 files changed

+50
-0
lines changed

2 files changed

+50
-0
lines changed

external/mmdetection/configs/custom-counting-instance-seg/efficientnetb2b_maskrcnn/model.py

+1
Original file line numberDiff line numberDiff line change
@@ -157,3 +157,4 @@
157157
dict(type='EarlyStoppingHook', patience=10, metric='mAP',
158158
interval=1, priority=75, iteration_patience=0),
159159
]
160+
fp16 = dict(loss_scale=512.)

external/mmdetection/tests/ote_cli/test_instance_segmentation.py

+49
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,13 @@
3434
ote_eval_testing,
3535
ote_train_testing,
3636
ote_export_testing,
37+
nncf_optimize_testing,
38+
nncf_export_testing,
39+
nncf_eval_testing,
40+
ote_hpo_testing,
41+
nncf_eval_openvino_testing,
42+
pot_optimize_testing,
43+
pot_eval_testing
3744
)
3845

3946

@@ -111,3 +118,45 @@ def test_ote_eval_deployment(self, template):
111118
@pytest.mark.parametrize("template", templates, ids=templates_ids)
112119
def test_ote_demo_deployment(self, template):
113120
ote_demo_deployment_testing(template, root, ote_dir, args)
121+
122+
@e2e_pytest_component
123+
@pytest.mark.parametrize("template", templates, ids=templates_ids)
124+
def test_nncf_optimize(self, template):
125+
if template.entrypoints.nncf is None:
126+
pytest.skip("nncf entrypoint is none")
127+
128+
nncf_optimize_testing(template, root, ote_dir, args)
129+
130+
@e2e_pytest_component
131+
@pytest.mark.parametrize("template", templates, ids=templates_ids)
132+
def test_nncf_export(self, template):
133+
if template.entrypoints.nncf is None:
134+
pytest.skip("nncf entrypoint is none")
135+
136+
nncf_export_testing(template, root)
137+
138+
@e2e_pytest_component
139+
@pytest.mark.parametrize("template", templates, ids=templates_ids)
140+
def test_nncf_eval(self, template):
141+
if template.entrypoints.nncf is None:
142+
pytest.skip("nncf entrypoint is none")
143+
144+
nncf_eval_testing(template, root, ote_dir, args, threshold=0.001)
145+
146+
@e2e_pytest_component
147+
@pytest.mark.parametrize("template", templates, ids=templates_ids)
148+
def test_nncf_eval_openvino(self, template):
149+
if template.entrypoints.nncf is None:
150+
pytest.skip("nncf entrypoint is none")
151+
152+
nncf_eval_openvino_testing(template, root, ote_dir, args)
153+
154+
@e2e_pytest_component
155+
@pytest.mark.parametrize("template", templates, ids=templates_ids)
156+
def test_pot_optimize(self, template):
157+
pot_optimize_testing(template, root, ote_dir, args)
158+
159+
@e2e_pytest_component
160+
@pytest.mark.parametrize("template", templates, ids=templates_ids)
161+
def test_pot_eval(self, template):
162+
pot_eval_testing(template, root, ote_dir, args)

0 commit comments

Comments
 (0)