Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
c07c5de1
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
接近 2 年 前同步成功
通知
116
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看板
未验证
提交
c07c5de1
编写于
1月 25, 2021
作者:
jm_12138
提交者:
GitHub
1月 25, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix a bug of DropPath (#575)
上级
30fee0f9
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
3 addition
and
4 deletion
+3
-4
ppcls/modeling/architectures/vision_transformer.py
ppcls/modeling/architectures/vision_transformer.py
+3
-4
未找到文件。
ppcls/modeling/architectures/vision_transformer.py
浏览文件 @
c07c5de1
...
@@ -42,14 +42,13 @@ def drop_path(x, drop_prob=0., training=False):
...
@@ -42,14 +42,13 @@ def drop_path(x, drop_prob=0., training=False):
"""
"""
if
drop_prob
==
0.
or
not
training
:
if
drop_prob
==
0.
or
not
training
:
return
x
return
x
keep_prob
=
1
-
drop_prob
keep_prob
=
paddle
.
to_tensor
(
1
-
drop_prob
)
shape
=
(
x
.
shape
[
0
],)
+
(
1
,)
*
(
x
.
ndim
-
1
)
shape
=
(
x
.
shape
[
0
],)
+
(
1
,)
*
(
x
.
ndim
-
1
)
random_tensor
=
keep_prob
+
paddle
.
rand
(
shape
,
dtype
=
x
.
dtype
)
random_tensor
=
keep_prob
+
paddle
.
rand
(
shape
,
dtype
=
x
.
dtype
)
random_tensor
.
floor_
()
# binarize
random_tensor
=
paddle
.
floor
(
random_tensor
)
# binarize
output
=
x
.
div
(
keep_prob
)
*
random_tensor
output
=
x
.
div
ide
(
keep_prob
)
*
random_tensor
return
output
return
output
class
DropPath
(
nn
.
Layer
):
class
DropPath
(
nn
.
Layer
):
"""Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
"""Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
"""
"""
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录