diff --git a/dygraph/paddleseg/models/ann.py b/dygraph/paddleseg/models/ann.py
index d2fd3688c48425f978612250d7a7d261cda752b6..a9d805a5ed02aef4326191b42c04745ae515867e 100644
--- a/dygraph/paddleseg/models/ann.py
+++ b/dygraph/paddleseg/models/ann.py
@@ -18,7 +18,7 @@ import paddle
 import paddle.nn.functional as F
 from paddle import nn
 
-from paddleseg.cvlibs import manager
+from paddleseg.cvlibs import manager, param_init
 from paddleseg.models.common.layer_libs import ConvBNReLU, ConvBN, AuxLayer
 from paddleseg.utils import utils
 
@@ -294,7 +294,7 @@ class SelfAttentionBlock_AFNB(nn.Layer):
         key_channels (int): the key channels in self-attention block.
         value_channels (int): the value channels in self-attention block.
         out_channels (int): out channels of AFNB module.
-        scale (int): pooling size. Defaut to 1.
+        scale (int): pooling size. Default to 1.
         psp_size (tuple): the out size of pooled feature maps. Default to (1, 3, 6, 8).
     """
 
@@ -374,7 +374,7 @@ class SelfAttentionBlock_APNB(nn.Layer):
         out_channels (int): out channels of APNB module.
         key_channels (int): the key channels in self-attention block.
         value_channels (int): the value channels in self-attention block.
-        scale (int): pooling size. Defaut to 1.
+        scale (int): pooling size. Default to 1.
         psp_size (tuple): the out size of pooled feature maps. Default to (1, 3, 6, 8).
     """