From bec1289fcd2c53b8db8516bc1475a1ab2bb91674 Mon Sep 17 00:00:00 2001
From: jm12138 <2286040843@qq.com>
Date: Thu, 10 Jun 2021 23:20:29 +0800
Subject: [PATCH] update __all__ export

---
 ppcls/arch/backbone/model_zoo/hardnet.py | 3 +++
 ppcls/arch/backbone/model_zoo/rednet.py  | 3 +++
 ppcls/arch/backbone/model_zoo/tnt.py     | 3 +++
 3 files changed, 9 insertions(+)

diff --git a/ppcls/arch/backbone/model_zoo/hardnet.py b/ppcls/arch/backbone/model_zoo/hardnet.py
index 006db8562..13651e794 100644
--- a/ppcls/arch/backbone/model_zoo/hardnet.py
+++ b/ppcls/arch/backbone/model_zoo/hardnet.py
@@ -16,6 +16,9 @@ MODEL_URLS = {
 }
 
 
+__all__ = MODEL_URLS.keys()
+
+
 def ConvLayer(in_channels, out_channels, kernel_size=3, stride=1, bias_attr=False):
     layer = nn.Sequential(
         ('conv', nn.Conv2D(
diff --git a/ppcls/arch/backbone/model_zoo/rednet.py b/ppcls/arch/backbone/model_zoo/rednet.py
index c3e037aa6..d80ea5e04 100644
--- a/ppcls/arch/backbone/model_zoo/rednet.py
+++ b/ppcls/arch/backbone/model_zoo/rednet.py
@@ -20,6 +20,9 @@ MODEL_URLS = {
 }
 
 
+__all__ = MODEL_URLS.keys()
+
+
 class Involution(nn.Layer):
     def __init__(self, channels, kernel_size, stride):
         super(Involution, self).__init__()
diff --git a/ppcls/arch/backbone/model_zoo/tnt.py b/ppcls/arch/backbone/model_zoo/tnt.py
index f95a4bcdc..2bd31566d 100644
--- a/ppcls/arch/backbone/model_zoo/tnt.py
+++ b/ppcls/arch/backbone/model_zoo/tnt.py
@@ -16,6 +16,9 @@ MODEL_URLS = {
 }
 
 
+__all__ = MODEL_URLS.keys()
+
+
 trunc_normal_ = TruncatedNormal(std=.02)
 zeros_ = Constant(value=0.)
 ones_ = Constant(value=1.)