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

Co-authored-by: woobinw <Wubin.Wu@imgtec.com>
daobook · Nov 25, 2022 · 9ca348b · 9ca348b
1 parent 72f3e7a
commit 9ca348b
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)