Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
cca06106
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看板
提交
cca06106
编写于
5月 15, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
5月 15, 2020
浏览文件
操作
浏览文件
下载
差异文件
!1158 fix some pyling warnings
Merge pull request !1158 from SJN/master
上级
d2efa20c
f2230fb0
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
7 addition
and
1 deletion
+7
-1
example/bert_clue/evaluation.py
example/bert_clue/evaluation.py
+2
-0
example/bert_clue/finetune.py
example/bert_clue/finetune.py
+2
-0
mindspore/nn/layer/quant.py
mindspore/nn/layer/quant.py
+1
-0
mindspore/ops/operations/nn_ops.py
mindspore/ops/operations/nn_ops.py
+1
-1
mindspore/train/summary/_event_writer.py
mindspore/train/summary/_event_writer.py
+1
-0
未找到文件。
example/bert_clue/evaluation.py
浏览文件 @
cca06106
...
...
@@ -82,6 +82,8 @@ def get_dataset(batch_size=1, repeat_count=1, distribute_file=''):
'''
get dataset
'''
_
=
distribute_file
ds
=
de
.
TFRecordDataset
([
cfg
.
data_file
],
cfg
.
schema_file
,
columns_list
=
[
"input_ids"
,
"input_mask"
,
"segment_ids"
,
"label_ids"
])
type_cast_op
=
C
.
TypeCast
(
mstype
.
int32
)
...
...
example/bert_clue/finetune.py
浏览文件 @
cca06106
...
...
@@ -58,6 +58,8 @@ def get_dataset(batch_size=1, repeat_count=1, distribute_file=''):
'''
get dataset
'''
_
=
distribute_file
ds
=
de
.
TFRecordDataset
([
cfg
.
data_file
],
cfg
.
schema_file
,
columns_list
=
[
"input_ids"
,
"input_mask"
,
"segment_ids"
,
"label_ids"
])
type_cast_op
=
C
.
TypeCast
(
mstype
.
int32
)
...
...
mindspore/nn/layer/quant.py
浏览文件 @
cca06106
...
...
@@ -208,6 +208,7 @@ class Conv2dBatchNormQuant(Cell):
symmetric
=
False
,
narrow_range
=
False
):
super
(
Conv2dBatchNormQuant
,
self
).
__init__
()
_
=
dilation
self
.
stride
=
stride
self
.
conv
=
P
.
Conv2D
(
out_channel
=
out_channels
,
kernel_size
=
kernel_size
,
...
...
mindspore/ops/operations/nn_ops.py
浏览文件 @
cca06106
...
...
@@ -2030,7 +2030,7 @@ class OneHot(PrimitiveWithInfer):
depth_val
=
depth
[
'value'
]
validator
.
check_integer
(
"depth"
,
depth_val
,
0
,
Rel
.
GE
,
self
.
name
)
# create new dimension at end if self.axis is -1
indices_shp
.
insert
(
self
.
axis
,
depth_val
)
if
self
.
axis
>=
0
else
indices_shp
.
append
(
depth_val
)
_
=
indices_shp
.
insert
(
self
.
axis
,
depth_val
)
if
self
.
axis
>=
0
else
indices_shp
.
append
(
depth_val
)
return
{
'shape'
:
indices_shp
,
'dtype'
:
on_value
[
'dtype'
],
...
...
mindspore/train/summary/_event_writer.py
浏览文件 @
cca06106
...
...
@@ -38,6 +38,7 @@ class EventWriter(Process):
def
__init__
(
self
,
filepath
:
str
,
flush_interval
:
int
)
->
None
:
super
().
__init__
()
_
=
flush_interval
with
open
(
filepath
,
'w'
):
os
.
chmod
(
filepath
,
stat
.
S_IWUSR
|
stat
.
S_IRUSR
)
self
.
_writer
=
EventWriter_
(
filepath
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录