diff --git a/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py b/x2paddle/op_mapper/caffe2paddle/caffe_custom_layer/priorbox.py index ee3fe16c54d4d3a956ba527f32e7ce8e585ba208..ecdb892b6915d0f57720d7998b998e12d6f6b039 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 456605bc8812187ff6f252ba41c2b7c10e130c9b..e99906bd0a769e7373bcf3f36d8424c9bf97ada8 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,