From 4511b8793f1343d5f1d22143f63d53449409c005 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Thu, 7 Jan 2021 16:24:06 +0800
Subject: [PATCH] fix hrnet comment (#2020)

---
 ppdet/modeling/backbones/hrnet.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ppdet/modeling/backbones/hrnet.py b/ppdet/modeling/backbones/hrnet.py
index 2849d942a..bd86fb79b 100644
--- a/ppdet/modeling/backbones/hrnet.py
+++ b/ppdet/modeling/backbones/hrnet.py
@@ -39,12 +39,12 @@ class HRNet(object):
     """
     HRNet, see https://arxiv.org/abs/1908.07919
     Args:
-        depth (int): ResNet depth, should be 18, 34, 50, 101, 152.
+        width (int): network width, should be 18, 30, 32, 40, 44, 48, 60 or 64
+        has_se (bool): whether contain squeeze_excitation(SE) block or not
         freeze_at (int): freeze the backbone at which stage
-        norm_type (str): normalization type, 'bn'/'sync_bn'/'affine_channel'
+        norm_type (str): normalization type, 'bn'/'sync_bn'
         freeze_norm (bool): freeze normalization layers
         norm_decay (float): weight decay for normalization layer weights
-        variant (str): ResNet variant, supports 'a', 'b', 'c', 'd' currently
         feature_maps (list): index of stages whose feature maps are returned
     """
 
-- 
GitLab