fix bug: KeyError, can't find some parameter key (#12211)

Co-authored-by: woobinw <Wubin.Wu@imgtec.com>
apache · Jul 31, 2022 · 66a1b9f · 66a1b9f
1 parent 42dd6af
commit 66a1b9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)