diff --git a/python/paddle/fluid/imperative/nn.py b/python/paddle/fluid/imperative/nn.py index 4786f8b8ad3cdd3e16a5fb4ed15c32704f5c7990..5aff3ea2d1f45a3ad0290a649bea06143bc1f146 100644 --- a/python/paddle/fluid/imperative/nn.py +++ b/python/paddle/fluid/imperative/nn.py @@ -205,7 +205,7 @@ class FC(layers.Layer): self._num_flatten_dims = num_flatten_dims self._dtype = dtype self._param_attr = param_attr - self._bias_attr = param_attr + self._bias_attr = bias_attr self._act = act def _build_once(self, input): @@ -219,10 +219,10 @@ class FC(layers.Layer): dtype=self._dtype, is_bias=False) - if self._param_attr: + if self._bias_attr: size = list([self._size]) self._b = self.create_parameter( - attr=self._param_attr, + attr=self._bias_attr, shape=size, dtype=self._dtype, is_bias=True)