def test_infer_twice_c2(self):
"""test single infer twice and check the value.
"""
self.yaml_config_name = sys._getframe().f_code.co_name + '.yaml'
self.yaml_content['mode'] = 'runner2'
self.run_yaml()
batch_auc1 = built_in.extract_value(self.out, self.batch_auc_re)
auc1 = built_in.extract_value(self.out, self.auc_re)
self.run_yaml()
batch_auc2 = built_in.extract_value(self.out, self.batch_auc_re)
auc2 = built_in.extract_value(self.out, self.auc_re)
err_msg = "{} != {}".format(batch_auc1, batch_auc2)
built_in.numpy_close(batch_auc1, batch_auc2, err_msg)
err_msg = "{} != {}".format(auc1, auc2)
built_in.numpy_close(auc1, auc2, err_msg)
def test_infer_twice(self):
"""test infer twice."""
self.yaml_config_name = sys._getframe().f_code.co_name + '.yaml'
self.yaml_content['phase'].pop()
self.yaml_content["mode"] = "runner2"
self.yaml_content['phase'].append({
'name': 'phase2',
'model': '{workspace}/model.py', # user-defined model
'dataset_name': 'dataset_infer', # select dataset by name
'thread_num': 2
})
self.run_yaml()
l1 = built_in.extract_value(self.out, r'.+,\sAUC_marital:\s\[(.+)\],')
self.run_yaml()
l2 = built_in.extract_value(self.out, r'.+,\sAUC_marital:\s\[(.+)\],')
err_msg = "{} != {}".format(l1, l2)
built_in.numpy_close(l1, l2, err_msg)
def test_infer_twice_gpu(self):
"""test infer twice with gpu."""
self.yaml_config_name = sys._getframe().f_code.co_name + '.yaml'
self.yaml_content['phase'].pop()
self.yaml_content["mode"] = "runner2"
self.yaml_content["runner"][2]["device"] = 'gpu'
self.yaml_content['phase'].append({
'name': 'phase2',
'model': '{workspace}/model.py', # user-defined model
'dataset_name': 'dataset_infer', # select dataset by name
'thread_num': 2
})
self.run_yaml()
l1 = built_in.extract_value(self.out, self.auc_re)
self.run_yaml()
l2 = built_in.extract_value(self.out, self.auc_re)
err_msg = "{} != {}".format(l1, l2)
built_in.numpy_close(l1, l2, err_msg)
def test_two_phase_infer(self):
"""test two infer in phase."""
self.yaml_config_name = sys._getframe().f_code.co_name + '.yaml'
self.yaml_content['phase'].pop()
self.yaml_content["mode"] = "runner2"
self.yaml_content['phase'].append({
'name': 'phase2',
'model': '{workspace}/model.py', # user-defined model
'dataset_name': 'dataset_infer', # select dataset by name
'thread_num': 2
})
self.yaml_content['phase'].append({
'name': 'phase2',
'model': '{workspace}/model.py', # user-defined model
'dataset_name': 'dataset_infer', # select dataset by name
'thread_num': 2
})
self.run_yaml()
total_list = built_in.extract_value(self.out, self.auc_re)
err_msg = "{} != {}".format(total_list[0:1], total_list[2:3])
built_in.numpy_close(total_list[0:1], total_list[2:3], err_msg)
