From d7621aea22b05a92d5432239bdb9aa4a0da8c3e4 Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Fri, 13 May 2022 17:44:56 +0800 Subject: [PATCH] fixed inf bug --- x2paddle/op_mapper/onnx2paddle/opset9/opset.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py index 23cc125..a5c6e1a 100755 --- a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py @@ -261,6 +261,8 @@ class OpSet9(): shape = node.out_shapes[0] if hasattr(node.weight, "shape") and len(node.weight.shape) == 0: + if node.weight == float('inf') or node.weight == float('-inf'): + node.weight = string(node.weight) self.paddle_graph.add_layer( "paddle.full", inputs={}, @@ -797,6 +799,8 @@ class OpSet9(): if len(value) == 1: value = value.tolist() value = value[0] + if value == float('inf') or value == float('-inf'): + value = string(value) self.paddle_graph.add_layer( "paddle.full", inputs={}, -- GitLab