update code

PaddlePaddle · Sep 20, 2024 · 79bc735 · 79bc735
1 parent da89a71
commit 79bc735
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/nn/layer/container.py b/python/paddle/nn/layer/container.py
@@ -338,10 +338,10 @@ class ParameterDict(Layer):
             ...         super().__init__()
             ...         # create ParameterDict with iterable Parameters
             ...         self.params = paddle.nn.ParameterDict(
-            ...             {f"t{i}": paddle.create_parameter(shape=[2, 2], dtype='float32') for i in range(num_stacked_param)})
+            ...             {'t' + str(i): paddle.create_parameter(shape=[2, 2], dtype='float32') for i in range(num_stacked_param)})
             ...
             ...     def forward(self, x):
-            ...         for i, (key, p) in enumerate(self.params.items()):
+            ...         for i, (key, p) in enumerate(self.params):
             ...             tmp = self._helper.create_variable_for_type_inference('float32')
             ...             self._helper.append_op(
             ...                 type="mul",