apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 08:37:06 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix wrong api in fastdeploy model

Browse Source
This commit is contained in:
jiangjiajun
2022-11-22 06:25:05 +00:00
parent 1a9a2df782
commit 5d37f739e8
2 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
python/fastdeploy/model.py
View File

@@ -27,23 +27,23 @@ class FastDeployModel:
def model_name(self):
return self._model.model_name()
def num_inputs(self):
return self._model.num_inputs()
def num_inputs_of_runtime(self):
return self._model.num_inputs_of_runtime()
def num_outputs(self):
return self._model.num_outputs()
def num_outputs_of_runtime(self):
return self._model.num_outputs_of_runtime()
def get_input_info(self, index):
assert index < self.num_inputs(
def input_info_of_runtime(self, index):
assert index < self.num_inputs_of_runtime(
), "The index:{} must be less than number of inputs:{}.".format(
index, self.num_inputs())
return self._model.get_input_info(index)
index, self.num_inputs_of_runtime())
return self._model.input_info_of_runtime(index)
def get_output_info(self, index):
assert index < self.num_outputs(
def output_info_of_runtime(self, index):
assert index < self.num_outputs_of_runtime(
), "The index:{} must be less than number of outputs:{}.".format(
index, self.num_outputs())
return self._model.get_output_info(index)
index, self.num_outputs_of_runtime())
return self._model.output_info_of_runtime(index)
def enable_record_time_of_runtime(self):
self._model.enable_record_time_of_runtime()