mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

v2.6.2 dont support to pass directory of model files

pull/3313/head
gaotingquan 2024-12-10 07:23:20 +00:00
parent 3c49c927e2
commit 3be5c3304f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deploy/utils/predictor.py
View File

@ -57,7 +57,7 @@ class Predictor(object):
# NOTE: paddle support to PIR mode after v2.6.0
major_v, minor_v, _ = paddle.__version__.split(".")[:3]
major_v, minor_v = int(major_v), int(minor_v)
if (major_v == 0 and minor_v == 0) or (major_v == 2 and minor_v >= 6) or (major_v >= 3):
if (major_v == 0 and minor_v == 0) or (major_v >= 3):
config = Config(inference_model_dir, model_prefix)
else:
model_file = os.path.join(inference_model_dir, f"{model_prefix}.pdmodel")