From 2b2e5bced37df0f09dc47417c06ea7c7c45f1e8c Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Tue, 7 Jun 2022 12:58:12 +0800 Subject: [PATCH] fixed for ci --- x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py | 2 ++ .../op_mapper/caffe2paddle/caffe_custom_layer/roipooling.py | 1 + 2 files changed, 3 insertions(+) diff --git a/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py b/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py index ee3fe16..ecdb892 100644 --- a/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py +++ b/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py @@ -14,9 +14,11 @@ import paddle from paddle import _C_ops +from paddle import in_dynamic_mode from paddle.common_ops_import import Variable, LayerHelper, check_variable_and_dtype, check_type, check_dtype +@paddle.jit.not_to_static def prior_box(input, image, min_sizes, diff --git a/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/roipooling.py b/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/roipooling.py index 456605b..e99906b 100644 --- a/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/roipooling.py +++ b/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/roipooling.py @@ -18,6 +18,7 @@ from paddle import in_dynamic_mode from paddle.common_ops_import import Variable, LayerHelper, check_variable_and_dtype, check_type, check_dtype +@paddle.jit.not_to_static def roi_pool(input, rois, pooled_height, -- GitLab