Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
44d60bd9
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
44d60bd9
编写于
5月 19, 2017
作者:
H
Helin Wang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add comments for exported functions
上级
27fdccc3
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
13 addition
and
2 deletion
+13
-2
paddle/go/pserver/service.go
paddle/go/pserver/service.go
+12
-1
paddle/go/pserver/service_test.go
paddle/go/pserver/service_test.go
+1
-1
未找到文件。
paddle/go/pserver/service.go
浏览文件 @
44d60bd9
...
...
@@ -37,6 +37,7 @@ type ParameterWithConfig struct {
// Gradient is the gradient of the parameter.
type
Gradient
Parameter
// Service is the RPC service for pserver.
type
Service
struct
{
initialized
chan
struct
{}
...
...
@@ -45,6 +46,7 @@ type Service struct {
paramMap
map
[
string
]
Parameter
}
// NewService creates a new service.
func
NewService
()
*
Service
{
s
:=
&
Service
{}
s
.
paramMap
=
make
(
map
[
string
]
Parameter
)
...
...
@@ -52,6 +54,8 @@ func NewService() *Service {
return
s
}
// BeginInitParams tells the parameter server that the parameter
// initialization has begun.
func
(
s
*
Service
)
BeginInitParams
(
config
[]
byte
,
dummy
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
...
...
@@ -71,6 +75,7 @@ func (s *Service) BeginInitParams(config []byte, dummy *int) error {
return
nil
}
// InitParam initializes a parameter.
func
(
s
*
Service
)
InitParam
(
paramWithConfigs
ParameterWithConfig
,
dummy
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
...
...
@@ -90,6 +95,8 @@ func (s *Service) InitParam(paramWithConfigs ParameterWithConfig, dummy *int) er
return
nil
}
// FinishInitParams tells the parameter server that the parameter
// initialization has finished.
func
(
s
*
Service
)
FinishInitParams
(
dummy0
int
,
dummy1
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
...
...
@@ -101,6 +108,8 @@ func (s *Service) FinishInitParams(dummy0 int, dummy1 *int) error {
return
nil
}
// SendGrads sends gradients to parameter servers for parameter
// optimization.
func
(
s
*
Service
)
SendGrads
(
grads
[]
Gradient
,
dummy
*
int
)
error
{
<-
s
.
initialized
...
...
@@ -140,6 +149,7 @@ func (s *Service) SendGrads(grads []Gradient, dummy *int) error {
return
nil
}
// GetParams gets parameters from the parameter server.
func
(
s
*
Service
)
GetParams
(
names
[]
string
,
parameters
*
[]
Parameter
)
error
{
<-
s
.
initialized
s
.
mu
.
Lock
()
...
...
@@ -166,7 +176,8 @@ func (s *Service) GetParams(names []string, parameters *[]Parameter) error {
return
nil
}
func
(
s
*
Service
)
SaveModel
(
path
string
,
dummy
*
int
)
error
{
// Save tells the parameter server to save parameters.
func
(
s
*
Service
)
Save
(
path
string
,
dummy
*
int
)
error
{
<-
s
.
initialized
// TODO
...
...
paddle/go/pserver/service_test.go
浏览文件 @
44d60bd9
...
...
@@ -124,7 +124,7 @@ func TestBlockUntilInitialized(t *testing.T) {
wg
.
Add
(
1
)
go
func
()
{
var
dummy
int
err
:=
s
.
Save
Model
(
""
,
&
dummy
)
err
:=
s
.
Save
(
""
,
&
dummy
)
if
err
!=
nil
{
t
.
FailNow
()
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录