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)