Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
272e7536
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看板
提交
272e7536
编写于
1月 06, 2023
作者:
weixin_46524038
提交者:
cuicheng01
1月 20, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
modify swinv2
上级
7d573e5d
变更
3
展开全部
隐藏空白更改
内联
并排
Showing
3 changed file
with
990 addition
and
2 deletion
+990
-2
docs/zh_CN/models/ImageNet1k/SwinTransformerV2.md
docs/zh_CN/models/ImageNet1k/SwinTransformerV2.md
+1
-1
ppcls/arch/backbone/__init__.py
ppcls/arch/backbone/__init__.py
+1
-1
ppcls/arch/backbone/model_zoo/swin_transformer_v2.py
ppcls/arch/backbone/model_zoo/swin_transformer_v2.py
+988
-0
未找到文件。
docs/zh_CN/models/ImageNet1k/SwinTransformerV2.md
浏览文件 @
272e7536
...
...
@@ -72,7 +72,7 @@ SwinTransformerV2 在 SwinTransformer 的基础上进行改进,可处理大尺
## 3. 模型训练、评估和预测
此部分内容包括训练环境配置、ImageNet数据的准备、SwinTransformer
在 ImageNet 上的训练、评估、预测等内容。在
`ppcls/configs/ImageNet/SwinTransformer/`
中提供了 SwinTransformer
的训练配置,可以通过如下脚本启动训练:此部分内容可以参考
[
ResNet50 模型训练、评估和预测
](
./ResNet.md#3
)
。
此部分内容包括训练环境配置、ImageNet数据的准备、SwinTransformer
V2 在 ImageNet 上的训练、评估、预测等内容。在
`ppcls/configs/ImageNet/SwinTransformerV2/`
中提供了 SwinTransformerV2
的训练配置,可以通过如下脚本启动训练:此部分内容可以参考
[
ResNet50 模型训练、评估和预测
](
./ResNet.md#3
)
。
**备注:**
由于 SwinTransformer 系列模型默认使用的 GPU 数量为 8 个,所以在训练时,需要指定8个GPU,如
`python3 -m paddle.distributed.launch --gpus="0,1,2,3,4,5,6,7" tools/train.py -c xxx.yaml`
, 如果使用 4 个 GPU 训练,默认学习率需要减小一半,精度可能有损。
...
...
ppcls/arch/backbone/__init__.py
浏览文件 @
272e7536
...
...
@@ -54,7 +54,7 @@ from .model_zoo.regnet import RegNetX_200MF, RegNetX_400MF, RegNetX_600MF, RegNe
from
.model_zoo.vision_transformer
import
ViT_small_patch16_224
,
ViT_base_patch16_224
,
ViT_base_patch16_384
,
ViT_base_patch32_384
,
ViT_large_patch16_224
,
ViT_large_patch16_384
,
ViT_large_patch32_384
from
.model_zoo.distilled_vision_transformer
import
DeiT_tiny_patch16_224
,
DeiT_small_patch16_224
,
DeiT_base_patch16_224
,
DeiT_tiny_distilled_patch16_224
,
DeiT_small_distilled_patch16_224
,
DeiT_base_distilled_patch16_224
,
DeiT_base_patch16_384
,
DeiT_base_distilled_patch16_384
from
.legendary_models.swin_transformer
import
SwinTransformer_tiny_patch4_window7_224
,
SwinTransformer_small_patch4_window7_224
,
SwinTransformer_base_patch4_window7_224
,
SwinTransformer_base_patch4_window12_384
,
SwinTransformer_large_patch4_window7_224
,
SwinTransformer_large_patch4_window12_384
from
.model_zoo.swin_transformer
v2
import
SwinTransformerV2_tiny_patch4_window8_256
,
SwinTransformerV2_small_patch4_window8_256
,
SwinTransformerV2_base_patch4_window8_256
,
SwinTransformerV2_tiny_patch4_window16_256
,
SwinTransformerV2_small_patch4_window16_256
,
SwinTransformerV2_base_patch4_window16_256
,
SwinTransformerV2_base_patch4_window12to
16_256
,
SwinTransformerV2_base_patch4_window24_384
,
SwinTransformerV2_large_patch4_window16_256
,
SwinTransformerV2_large_patch4_window24_384
from
.model_zoo.swin_transformer
_v2
import
SwinTransformerV2_tiny_patch4_window8_256
,
SwinTransformerV2_small_patch4_window8_256
,
SwinTransformerV2_base_patch4_window8_256
,
SwinTransformerV2_tiny_patch4_window16_256
,
SwinTransformerV2_small_patch4_window16_256
,
SwinTransformerV2_base_patch4_window
16_256
,
SwinTransformerV2_base_patch4_window24_384
,
SwinTransformerV2_large_patch4_window16_256
,
SwinTransformerV2_large_patch4_window24_384
from
.model_zoo.cswin_transformer
import
CSWinTransformer_tiny_224
,
CSWinTransformer_small_224
,
CSWinTransformer_base_224
,
CSWinTransformer_large_224
,
CSWinTransformer_base_384
,
CSWinTransformer_large_384
from
.model_zoo.mixnet
import
MixNet_S
,
MixNet_M
,
MixNet_L
from
.model_zoo.rexnet
import
ReXNet_1_0
,
ReXNet_1_3
,
ReXNet_1_5
,
ReXNet_2_0
,
ReXNet_3_0
...
...
ppcls/arch/backbone/model_zoo/swin_transformer_v2.py
0 → 100644
浏览文件 @
272e7536
此差异已折叠。
点击以展开。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录