Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
橘蓝
Paddle
提交
d34516fb
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,发现更多精彩内容 >>
提交
d34516fb
编写于
9月 11, 2017
作者:
武
武毅
提交者:
GitHub
9月 11, 2017
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Get output when training (#3978)
* get output when training * follow comments
上级
0f7bc938
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
15 addition
and
4 deletion
+15
-4
python/paddle/v2/event.py
python/paddle/v2/event.py
+8
-2
python/paddle/v2/trainer.py
python/paddle/v2/trainer.py
+7
-2
未找到文件。
python/paddle/v2/event.py
浏览文件 @
d34516fb
...
...
@@ -53,10 +53,13 @@ class BeginPass(object):
class
EndPass
(
WithMetric
):
"""
Event On One Pass Training Complete.
To get the output of a specific layer, add "event.gm.getLayerOutputs('predict_layer')"
in your event_handler call back
"""
def
__init__
(
self
,
pass_id
,
evaluator
):
def
__init__
(
self
,
pass_id
,
evaluator
,
gm
):
self
.
pass_id
=
pass_id
self
.
gm
=
gm
WithMetric
.
__init__
(
self
,
evaluator
)
...
...
@@ -73,10 +76,13 @@ class BeginIteration(object):
class
EndIteration
(
WithMetric
):
"""
Event On One Batch Training Complete.
To get the output of a specific layer, add "event.gm.getLayerOutputs('predict_layer')"
in your event_handler call back
"""
def
__init__
(
self
,
pass_id
,
batch_id
,
cost
,
evaluator
):
def
__init__
(
self
,
pass_id
,
batch_id
,
cost
,
evaluator
,
gm
):
self
.
pass_id
=
pass_id
self
.
batch_id
=
batch_id
self
.
cost
=
cost
self
.
gm
=
gm
WithMetric
.
__init__
(
self
,
evaluator
)
python/paddle/v2/trainer.py
浏览文件 @
d34516fb
...
...
@@ -174,13 +174,18 @@ class SGD(object):
pass_id
=
pass_id
,
batch_id
=
batch_id
,
cost
=
cost
,
evaluator
=
batch_evaluator
))
evaluator
=
batch_evaluator
,
gm
=
self
.
__gradient_machine__
))
self
.
__parameter_updater__
.
finishBatch
(
cost
)
batch_evaluator
.
finish
()
self
.
__parameter_updater__
.
finishPass
()
pass_evaluator
.
finish
()
event_handler
(
v2_event
.
EndPass
(
pass_id
,
evaluator
=
pass_evaluator
))
event_handler
(
v2_event
.
EndPass
(
pass_id
,
evaluator
=
pass_evaluator
,
gm
=
self
.
__gradient_machine__
))
self
.
__gradient_machine__
.
finish
()
def
test
(
self
,
reader
,
feeding
=
None
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录