Skip to content

Commit 8a4a330

Browse files
author
channingss
committed
add prompt
1 parent 212c3d9 commit 8a4a330

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

x2paddle/decoder/onnx_decoder.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -464,10 +464,12 @@ def get_dynamic_shape_from_caffe2(self, layer, input_shapes):
464464
import torch
465465
version = torch.__version__
466466
if '1.1.0' not in version:
467-
print("torch==1.1.0 is required")
467+
print("your model have dynamic graph, torch==1.1.0 is required")
468468
return
469469
except:
470-
print("onnx is not installed, use \"pip install torch==1.1.0\".")
470+
print(
471+
"your model have dynamic graph, we use caff2 to inference graph, please use \"pip install torch==1.1.0\"."
472+
)
471473
return
472474
from caffe2.python.onnx.backend import prepare
473475
shape = input_shapes[0]

0 commit comments

Comments
 (0)