Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
兽拳
Paddle
提交
65afbe11
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,发现更多精彩内容 >>
提交
65afbe11
编写于
7月 03, 2017
作者:
D
dongzhihong
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
"fix gob register error"
上级
e6c98e49
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
18 addition
and
15 deletion
+18
-15
go/pserver/service.go
go/pserver/service.go
+18
-15
未找到文件。
go/pserver/service.go
浏览文件 @
65afbe11
...
...
@@ -25,7 +25,7 @@ const (
)
const
(
checkpoint_path
=
"/checkpoints/"
checkpoint_path
=
"
.
/checkpoints/"
)
// Supported element types
...
...
@@ -67,10 +67,10 @@ type Service struct {
optMap
map
[
string
]
*
optimizer
}
type
C
heckpoint
struct
{
u
uid
string
m
d5sum
string
t
imestamp
string
type
c
heckpoint
struct
{
U
uid
string
M
d5sum
string
T
imestamp
string
}
//serialize ParameterWithConfig to byte stream
...
...
@@ -93,6 +93,8 @@ func NewService(idx int) (*Service, error) {
}
s
.
optMap
=
make
(
map
[
string
]
*
optimizer
)
s
.
initialized
=
make
(
chan
struct
{})
gob
.
Register
(
ParameterWithConfig
{})
gob
.
Register
(
checkpoint
{})
return
s
,
nil
}
...
...
@@ -190,32 +192,33 @@ func (s *Service) Save(path string, dummy *int) error {
if
err
!=
nil
{
log
.
Errorln
(
err
)
}
ck
:=
C
heckpoint
{}
ck
:=
c
heckpoint
{}
h
:=
md5
.
New
()
ck
.
m
d5sum
=
hex
.
EncodeToString
(
h
.
Sum
(
content
))
ck
.
t
imestamp
=
time
.
Now
()
.
String
()
ck
.
u
uid
=
checkpoint_path
+
strconv
.
Itoa
(
s
.
idx
)
ck
.
M
d5sum
=
hex
.
EncodeToString
(
h
.
Sum
(
content
))
ck
.
T
imestamp
=
time
.
Now
()
.
String
()
ck
.
U
uid
=
checkpoint_path
+
strconv
.
Itoa
(
s
.
idx
)
ckbytes
,
err
:=
GetBytes
(
ck
)
if
err
!=
nil
{
log
.
Errorln
(
err
)
}
// TODO: according design doc, need to save
u
uid to etcd in json format
// {\"
uuid\": [UUID], \"md5\", \"MD5 sum\", \"t
imestamp\": xxxx}
// TODO: according design doc, need to save
U
uid to etcd in json format
// {\"
Uuid\": [UUID], \"md5\", \"MD5 sum\", \"T
imestamp\": xxxx}
log
.
Infof
(
"parameter checkpoint %s"
,
ckbytes
)
if
_
,
err
=
os
.
Stat
(
ck
.
u
uid
);
os
.
IsNotExist
(
err
)
{
if
_
,
err
=
os
.
Stat
(
ck
.
U
uid
);
os
.
IsNotExist
(
err
)
{
log
.
Info
(
"checkpoint not exists."
)
}
else
{
err
=
os
.
Remove
(
ck
.
u
uid
)
log
.
Infof
(
"remove %s"
,
ck
.
u
uid
)
err
=
os
.
Remove
(
ck
.
U
uid
)
log
.
Infof
(
"remove %s"
,
ck
.
U
uid
)
}
f
,
err
:=
os
.
Create
(
ck
.
u
uid
)
f
,
err
:=
os
.
Create
(
ck
.
U
uid
)
defer
f
.
Close
()
if
err
!=
nil
{
log
.
Errorln
(
err
)
}
writer
:=
bufio
.
NewWriter
(
f
)
_
,
err
=
writer
.
Write
(
content
)
writer
.
Flush
()
if
err
!=
nil
{
log
.
Errorln
(
err
)
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录