Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
4266b879
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
4266b879
编写于
7月 09, 2022
作者:
Y
Yang Nie
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
rename `ConvNext` to `ConvNeXt`
上级
436f64ad
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
5 addition
and
5 deletion
+5
-5
ppcls/arch/backbone/__init__.py
ppcls/arch/backbone/__init__.py
+1
-1
ppcls/arch/backbone/model_zoo/convnext.py
ppcls/arch/backbone/model_zoo/convnext.py
+3
-3
ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml
ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml
+1
-1
未找到文件。
ppcls/arch/backbone/__init__.py
浏览文件 @
4266b879
...
@@ -67,7 +67,7 @@ from ppcls.arch.backbone.model_zoo.pvt_v2 import PVT_V2_B0, PVT_V2_B1, PVT_V2_B2
...
@@ -67,7 +67,7 @@ from ppcls.arch.backbone.model_zoo.pvt_v2 import PVT_V2_B0, PVT_V2_B1, PVT_V2_B2
from
ppcls.arch.backbone.model_zoo.mobilevit
import
MobileViT_XXS
,
MobileViT_XS
,
MobileViT_S
from
ppcls.arch.backbone.model_zoo.mobilevit
import
MobileViT_XXS
,
MobileViT_XS
,
MobileViT_S
from
ppcls.arch.backbone.model_zoo.repvgg
import
RepVGG_A0
,
RepVGG_A1
,
RepVGG_A2
,
RepVGG_B0
,
RepVGG_B1
,
RepVGG_B2
,
RepVGG_B1g2
,
RepVGG_B1g4
,
RepVGG_B2g4
,
RepVGG_B3g4
from
ppcls.arch.backbone.model_zoo.repvgg
import
RepVGG_A0
,
RepVGG_A1
,
RepVGG_A2
,
RepVGG_B0
,
RepVGG_B1
,
RepVGG_B2
,
RepVGG_B1g2
,
RepVGG_B1g4
,
RepVGG_B2g4
,
RepVGG_B3g4
from
ppcls.arch.backbone.model_zoo.van
import
VAN_tiny
from
ppcls.arch.backbone.model_zoo.van
import
VAN_tiny
from
ppcls.arch.backbone.model_zoo.convnext
import
ConvNe
x
t_tiny
from
ppcls.arch.backbone.model_zoo.convnext
import
ConvNe
X
t_tiny
from
ppcls.arch.backbone.variant_models.resnet_variant
import
ResNet50_last_stage_stride1
from
ppcls.arch.backbone.variant_models.resnet_variant
import
ResNet50_last_stage_stride1
from
ppcls.arch.backbone.variant_models.vgg_variant
import
VGG19Sigmoid
from
ppcls.arch.backbone.variant_models.vgg_variant
import
VGG19Sigmoid
from
ppcls.arch.backbone.variant_models.pp_lcnet_variant
import
PPLCNet_x2_5_Tanh
from
ppcls.arch.backbone.variant_models.pp_lcnet_variant
import
PPLCNet_x2_5_Tanh
...
...
ppcls/arch/backbone/model_zoo/convnext.py
浏览文件 @
4266b879
...
@@ -29,7 +29,7 @@ from paddle.nn.initializer import TruncatedNormal, Constant
...
@@ -29,7 +29,7 @@ from paddle.nn.initializer import TruncatedNormal, Constant
from
ppcls.utils.save_load
import
load_dygraph_pretrain
,
load_dygraph_pretrain_from_url
from
ppcls.utils.save_load
import
load_dygraph_pretrain
,
load_dygraph_pretrain_from_url
MODEL_URLS
=
{
MODEL_URLS
=
{
"ConvNe
x
t_tiny"
:
""
,
# TODO
"ConvNe
X
t_tiny"
:
""
,
# TODO
}
}
__all__
=
list
(
MODEL_URLS
.
keys
())
__all__
=
list
(
MODEL_URLS
.
keys
())
...
@@ -233,8 +233,8 @@ def _load_pretrained(pretrained, model, model_url, use_ssld=False):
...
@@ -233,8 +233,8 @@ def _load_pretrained(pretrained, model, model_url, use_ssld=False):
)
)
def
ConvNe
x
t_tiny
(
pretrained
=
False
,
use_ssld
=
False
,
**
kwargs
):
def
ConvNe
X
t_tiny
(
pretrained
=
False
,
use_ssld
=
False
,
**
kwargs
):
model
=
ConvNeXt
(
depths
=
[
3
,
3
,
9
,
3
],
dims
=
[
96
,
192
,
384
,
768
],
**
kwargs
)
model
=
ConvNeXt
(
depths
=
[
3
,
3
,
9
,
3
],
dims
=
[
96
,
192
,
384
,
768
],
**
kwargs
)
_load_pretrained
(
_load_pretrained
(
pretrained
,
model
,
MODEL_URLS
[
"ConvNe
x
t_tiny"
],
use_ssld
=
use_ssld
)
pretrained
,
model
,
MODEL_URLS
[
"ConvNe
X
t_tiny"
],
use_ssld
=
use_ssld
)
return
model
return
model
ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml
浏览文件 @
4266b879
...
@@ -24,7 +24,7 @@ EMA:
...
@@ -24,7 +24,7 @@ EMA:
# model architecture
# model architecture
Arch
:
Arch
:
name
:
ConvNe
x
t_tiny
name
:
ConvNe
X
t_tiny
class_num
:
1000
class_num
:
1000
drop_path_rate
:
0.1
drop_path_rate
:
0.1
layer_scale_init_value
:
1e-6
layer_scale_init_value
:
1e-6
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录