Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
正统之独孤求败
mindspore
提交
4bbd4414
M
mindspore
项目概览
正统之独孤求败
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
0
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
4bbd4414
编写于
6月 22, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
6月 22, 2020
浏览文件
操作
浏览文件
下载
差异文件
!1734 deal with resnet50_THOR train print many ERROR&WARNING log and produce many ir file
Merge pull request !1734 from zongha/r0.3
上级
cf7c60a5
491ba51b
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
8 addition
and
2 deletion
+8
-2
example/resnet50_imagenet2012_THOR/train.py
example/resnet50_imagenet2012_THOR/train.py
+1
-1
mindspore/ccsrc/utils/convert_utils.cc
mindspore/ccsrc/utils/convert_utils.cc
+1
-1
mindspore/ops/operations/thor_ops.py
mindspore/ops/operations/thor_ops.py
+6
-0
未找到文件。
example/resnet50_imagenet2012_THOR/train.py
浏览文件 @
4bbd4414
...
...
@@ -46,7 +46,7 @@ parser.add_argument('--dataset_path', type=str, default=None, help='Dataset path
args_opt
=
parser
.
parse_args
()
device_id
=
int
(
os
.
getenv
(
'DEVICE_ID'
))
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
,
save_graphs
=
Tru
e
,
device_id
=
device_id
)
context
.
set_context
(
mode
=
context
.
GRAPH_MODE
,
device_target
=
"Ascend"
,
save_graphs
=
Fals
e
,
device_id
=
device_id
)
def
get_model_lr
(
global_step
,
lr_init
,
decay
,
total_epochs
,
steps_per_epoch
):
...
...
mindspore/ccsrc/utils/convert_utils.cc
浏览文件 @
4bbd4414
...
...
@@ -121,7 +121,7 @@ py::object ValuePtrToPyData(const ValuePtr &value) {
}
else
if
(
value
->
isa
<
None
>
())
{
ret
=
py
::
none
();
}
else
{
MS_LOG
(
EXCEPTION
)
<<
"Unsupported convert value: "
<<
value
->
ToString
()
<<
" to a PyData."
;
MS_LOG
(
INFO
)
<<
"Unsupported convert value: "
<<
value
->
ToString
()
<<
" to a PyData."
;
}
return
ret
;
}
...
...
mindspore/ops/operations/thor_ops.py
浏览文件 @
4bbd4414
...
...
@@ -96,6 +96,12 @@ class CusCholeskyTrsm(PrimitiveWithInfer):
"""init CusCholeskyTrsm"""
self
.
init_prim_io_names
(
inputs
=
[
'x1'
],
outputs
=
[
'y'
])
from
mindspore.ops._op_impl._custom_op.cholesky_trsm_impl
import
CusCholeskyTrsm
def
get_bprop
(
self
):
def
bprop
(
x
,
out
,
dout
):
return
(
C
.
zeros_like
(
x
),)
return
bprop
def
infer_shape
(
self
,
data1_shape
):
ll
=
[]
m
,
_
=
data1_shape
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录