From 84e7ad5db9896fc20c7038588157a8108f35387f Mon Sep 17 00:00:00 2001 From: michaelowenliu Date: Sun, 27 Sep 2020 20:48:37 +0800 Subject: [PATCH] fix a typo --- dygraph/paddleseg/models/ann.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dygraph/paddleseg/models/ann.py b/dygraph/paddleseg/models/ann.py index d2fd3688..a9d805a5 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). """ -- GitLab