Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
N
NiuTrans.Tensor
概览
Overview
Details
Activity
Cycle Analytics
版本库
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
问题
8
Issues
8
列表
Board
标记
里程碑
合并请求
0
Merge Requests
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
Snippets
成员
Collapse sidebar
Close sidebar
活动
图像
聊天
创建新问题
作业
提交
Issue Boards
Open sidebar
NiuTrans
NiuTrans.Tensor
Commits
69478776
Commit
69478776
authored
4 years ago
by
xiaotong
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
updates
parent
c2a40cf0
显示空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
3 行增加
和
6 行删除
+3
-6
source/train/XLeader.cpp
+3
-6
没有找到文件。
source/train/XLeader.cpp
查看文件 @
69478776
...
@@ -509,14 +509,11 @@ void XLeader::RunUpdate(XConfig * config, XOptimizer * optimizer, const int * ac
...
@@ -509,14 +509,11 @@ void XLeader::RunUpdate(XConfig * config, XOptimizer * optimizer, const int * ac
CheckNTErrors
(
jobQueues
.
count
==
serverModel
.
paramNum
,
"Incompatiable model!"
);
CheckNTErrors
(
jobQueues
.
count
==
serverModel
.
paramNum
,
"Incompatiable model!"
);
/* counts how many member models are collect
for each parameters
*/
/* counts how many member models are collect
ed for each parameter
*/
int
*
finishedCount
=
new
int
[
serverModel
.
paramNum
];
int
*
finishedCount
=
new
int
[
serverModel
.
paramNum
];
memset
(
finishedCount
,
0
,
sizeof
(
int
)
*
serverModel
.
paramNum
);
memset
(
finishedCount
,
0
,
sizeof
(
int
)
*
serverModel
.
paramNum
);
/* This is a simple implementation of the wait-and-collect process. But
/* This is a simple implementation of the do-and-wait process */
there is a risk that some models are not available, that is, the
loop would never stop. A solution might be that we force the loop
to break after waiting for a short time. */
while
(
1
)
{
while
(
1
)
{
for
(
int
j
=
0
;
j
<
serverModel
.
paramNum
;
j
++
)
{
for
(
int
j
=
0
;
j
<
serverModel
.
paramNum
;
j
++
)
{
...
@@ -569,7 +566,7 @@ void XLeader::RunUpdate(XConfig * config, XOptimizer * optimizer, const int * ac
...
@@ -569,7 +566,7 @@ void XLeader::RunUpdate(XConfig * config, XOptimizer * optimizer, const int * ac
}
}
}
}
/*
the collection
finishes if all data tensors are processed */
/* finishes if all data tensors are processed */
if
(
finished
==
serverModel
.
paramNum
*
members
.
count
)
if
(
finished
==
serverModel
.
paramNum
*
members
.
count
)
break
;
break
;
...
...
This diff is collapsed.
Click to expand it.
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论