diff --git a/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py b/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
index 118d998992ec49176e40f5f4b74b20a0a6abd9fe..04b9557059ddd75faa89d6f9b013d2f4efb59ef4 100644
--- a/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
+++ b/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
@@ -138,7 +138,7 @@ class PyTorchOpMapper(OpMapper):
         # 更新split参数
         for layer in graph.layers.values():
             if layer.kernel == "paddle.split" and "num_or_sections" in layer.attrs \
-            and len(set(layer.attrs["num_or_sections"])) == 1:
+            and not isinstance(layer.attrs["num_or_sections"], int) and len(set(layer.attrs["num_or_sections"])) == 1:
                 layer.attrs["num_or_sections"] = self.split_len[layer.outputs[
                     0]]
         return graph, graph_inputs