Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
PaddleRec
提交
3d2c41e1
P
PaddleRec
项目概览
BaiXuePrincess
/
PaddleRec
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleRec
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleRec
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
3d2c41e1
编写于
9月 12, 2019
作者:
X
xiexionghang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix xbox-base-dump && add checkpoint config
上级
57882670
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
2 deletion
+4
-2
paddle/fluid/train/custom_trainer/feed/accessor/epoch_accessor.cc
...luid/train/custom_trainer/feed/accessor/epoch_accessor.cc
+3
-2
paddle/fluid/train/custom_trainer/feed/accessor/epoch_accessor.h
...fluid/train/custom_trainer/feed/accessor/epoch_accessor.h
+1
-0
未找到文件。
paddle/fluid/train/custom_trainer/feed/accessor/epoch_accessor.cc
浏览文件 @
3d2c41e1
...
@@ -96,6 +96,7 @@ namespace feed {
...
@@ -96,6 +96,7 @@ namespace feed {
std
::
shared_ptr
<
TrainerContext
>
context_ptr
)
{
std
::
shared_ptr
<
TrainerContext
>
context_ptr
)
{
_time_zone_seconds
=
config
[
"time_zone_seconds"
].
as
<
int
>
();
_time_zone_seconds
=
config
[
"time_zone_seconds"
].
as
<
int
>
();
_train_time_interval
=
config
[
"train_time_interval"
].
as
<
int
>
();
_train_time_interval
=
config
[
"train_time_interval"
].
as
<
int
>
();
_checkpoint_time_interval
=
config
[
"checkpoint_time_interval"
].
as
<
int
>
(
3600
*
18
);
// 默认每18小时dump一个
CHECK
(
_train_time_interval
>
0
&&
(
_train_time_interval
%
SecondsPerMin
)
==
0
);
CHECK
(
_train_time_interval
>
0
&&
(
_train_time_interval
%
SecondsPerMin
)
==
0
);
_train_num_per_day
=
SecondsPerDay
/
_train_time_interval
;
_train_num_per_day
=
SecondsPerDay
/
_train_time_interval
;
return
EpochAccessor
::
initialize
(
config
,
context_ptr
);
return
EpochAccessor
::
initialize
(
config
,
context_ptr
);
...
@@ -123,7 +124,7 @@ namespace feed {
...
@@ -123,7 +124,7 @@ namespace feed {
bool
TimelyEpochAccessor
::
is_last_epoch
(
uint64_t
epoch_id
)
{
bool
TimelyEpochAccessor
::
is_last_epoch
(
uint64_t
epoch_id
)
{
auto
delta
=
delta_id
(
epoch_id
);
auto
delta
=
delta_id
(
epoch_id
);
return
delta
==
_train_num_per_day
;
return
delta
==
0
;
// 最后一个delta恰好整除
}
}
uint64_t
TimelyEpochAccessor
::
epoch_time_interval
()
{
uint64_t
TimelyEpochAccessor
::
epoch_time_interval
()
{
...
@@ -149,7 +150,7 @@ namespace feed {
...
@@ -149,7 +150,7 @@ namespace feed {
if
(
is_last_epoch
(
epoch_id
))
{
if
(
is_last_epoch
(
epoch_id
))
{
return
true
;
return
true
;
}
}
return
delta_id
(
epoch_id
)
%
78
==
0
;
return
delta_id
(
epoch_id
)
%
(
_checkpoint_time_interval
/
_train_time_interval
)
==
0
;
case
ModelSaveWay
::
ModelSaveTrainCheckpointBase
:
case
ModelSaveWay
::
ModelSaveTrainCheckpointBase
:
return
is_last_epoch
(
epoch_id
);
return
is_last_epoch
(
epoch_id
);
}
}
...
...
paddle/fluid/train/custom_trainer/feed/accessor/epoch_accessor.h
浏览文件 @
3d2c41e1
...
@@ -103,6 +103,7 @@ private:
...
@@ -103,6 +103,7 @@ private:
uint32_t
_time_zone_seconds
;
// 相对UTC时差(秒)
uint32_t
_time_zone_seconds
;
// 相对UTC时差(秒)
uint32_t
_train_time_interval
;
// 训练时间间隔(秒)
uint32_t
_train_time_interval
;
// 训练时间间隔(秒)
uint32_t
_train_num_per_day
;
// 天级训练总轮数
uint32_t
_train_num_per_day
;
// 天级训练总轮数
uint32_t
_checkpoint_time_interval
;
// 每隔n秒,dump出CheckPoint
};
};
}
// namespace feed
}
// namespace feed
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录