Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
18f0af05
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
18f0af05
编写于
5月 25, 2020
作者:
“
“liuxiao”
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
pylint clean
上级
c8f69f5d
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
4 addition
and
4 deletion
+4
-4
tests/mindspore_test_framework/apps/bert_attention_submodules.py
...indspore_test_framework/apps/bert_attention_submodules.py
+2
-2
tests/st/networks/test_gpu_lstm.py
tests/st/networks/test_gpu_lstm.py
+1
-1
tests/ut/python/ir/test_tensor.py
tests/ut/python/ir/test_tensor.py
+1
-1
未找到文件。
tests/mindspore_test_framework/apps/bert_attention_submodules.py
浏览文件 @
18f0af05
...
...
@@ -167,7 +167,7 @@ class BertAttentionMask(nn.Cell):
super
(
BertAttentionMask
,
self
).
__init__
()
self
.
has_attention_mask
=
has_attention_mask
self
.
multiply_data
=
Tensor
([
-
1000.0
,
],
dtype
=
dtype
)
self
.
multiply_data
=
Tensor
([
-
1000.0
,],
dtype
=
dtype
)
self
.
multiply
=
P
.
Mul
()
if
self
.
has_attention_mask
:
...
...
@@ -198,7 +198,7 @@ class BertAttentionMaskBackward(nn.Cell):
dtype
=
mstype
.
float32
):
super
(
BertAttentionMaskBackward
,
self
).
__init__
()
self
.
has_attention_mask
=
has_attention_mask
self
.
multiply_data
=
Tensor
([
-
1000.0
,
],
dtype
=
dtype
)
self
.
multiply_data
=
Tensor
([
-
1000.0
,],
dtype
=
dtype
)
self
.
multiply
=
P
.
Mul
()
self
.
attention_mask
=
Tensor
(
np
.
ones
(
shape
=
attention_mask_shape
).
astype
(
np
.
float32
))
if
self
.
has_attention_mask
:
...
...
tests/st/networks/test_gpu_lstm.py
浏览文件 @
18f0af05
...
...
@@ -136,7 +136,7 @@ def test_LSTM():
train_network
.
set_train
()
train_features
=
Tensor
(
np
.
ones
([
64
,
max_len
]).
astype
(
np
.
int32
))
train_labels
=
Tensor
(
np
.
ones
([
64
,
]).
astype
(
np
.
int32
)[
0
:
64
])
train_labels
=
Tensor
(
np
.
ones
([
64
,]).
astype
(
np
.
int32
)[
0
:
64
])
losses
=
[]
for
epoch
in
range
(
num_epochs
):
loss
=
train_network
(
train_features
,
train_labels
)
...
...
tests/ut/python/ir/test_tensor.py
浏览文件 @
18f0af05
...
...
@@ -34,7 +34,7 @@ ndarr = np.ones((2, 3))
def
test_tensor_flatten
():
with
pytest
.
raises
(
AttributeError
):
lst
=
[
1
,
2
,
3
,
4
,
]
lst
=
[
1
,
2
,
3
,
4
,]
tensor_list
=
ms
.
Tensor
(
lst
,
ms
.
float32
)
tensor_list
=
tensor_list
.
Flatten
()
print
(
tensor_list
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录