From 9ca348b07e027dad75d0dea3a8c24a5f892351c1 Mon Sep 17 00:00:00 2001 From: woobinw <109947895+woobinw@users.noreply.github.com> Date: Sun, 31 Jul 2022 14:37:53 +0800 Subject: [PATCH] fix bug: KeyError, can't find some parameter key (#12211) Co-authored-by: woobinw --- python/tvm/relay/frontend/paddlepaddle.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/paddlepaddle.py b/python/tvm/relay/frontend/paddlepaddle.py index 7042154709ae..a869e2e1b807 100644 --- a/python/tvm/relay/frontend/paddlepaddle.py +++ b/python/tvm/relay/frontend/paddlepaddle.py @@ -2268,7 +2268,7 @@ def from_translated_layer(self, layer, shape_dict): self.shape_dict = shape_dict program = layer.program() parameters = dict() - for param in layer.parameters(): + for param in layer.parameters() + layer.buffers(): parameters[param.name] = np.array(param.value().get_tensor()) self.check_unsupported_ops(program) self.extract_parameters(program, parameters)