Rename pybind/fastdeploy_runtime.cc to pybind/runtime.cc (#273)

* fix yaml

* Rename pybind/fastdeploy_runtime.cc to runtime.cc
This commit is contained in:
Jason
2022-09-23 11:16:02 +08:00
committed by GitHub
parent efa7411ebb
commit 2d3e61582d
7 changed files with 11 additions and 9 deletions

View File

@@ -13,6 +13,7 @@
# limitations under the License.
import fastdeploy as fd
from fastdeploy import ModelFormat
import numpy as np
# 下载模型并解压
@@ -21,7 +22,7 @@ fd.download(model_url, path=".")
option = fd.RuntimeOption()
option.set_model_path("mobilenetv2.onnx", model_format="onnx")
option.set_model_path("mobilenetv2.onnx", model_format=ModelFormat.ONNX)
# **** GPU 配置 ***
option.use_gpu(0)