Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
31a04814
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
1 年多 前同步成功
通知
116
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
31a04814
编写于
11月 27, 2020
作者:
L
littletomatodonkey
提交者:
GitHub
11月 27, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix logger (#423)
* fix logger * fix typo
上级
82fceb5b
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
21 addition
and
26 deletion
+21
-26
tools/program.py
tools/program.py
+21
-26
未找到文件。
tools/program.py
浏览文件 @
31a04814
...
...
@@ -479,18 +479,24 @@ def run(dataloader,
metric_list
=
[
f
[
1
]
for
f
in
fetchs
.
values
()]
for
m
in
metric_list
:
m
.
reset
()
batch_time
=
AverageMeter
(
'elapse'
,
'.
3
f'
)
batch_time
=
AverageMeter
(
'elapse'
,
'.
5
f'
)
tic
=
time
.
time
()
dataloader
=
dataloader
if
config
.
get
(
'use_dali'
)
else
dataloader
()()
for
idx
,
batch
in
enumerate
(
dataloader
):
if
idx
==
10
:
for
m
in
metric_list
:
m
.
reset
()
batch_time
.
reset
()
batch_size
=
batch
[
0
][
"feed_image"
].
shape
()[
0
]
metrics
=
exe
.
run
(
program
=
program
,
feed
=
batch
,
fetch_list
=
fetch_list
)
batch_time
.
update
(
time
.
time
()
-
tic
)
tic
=
time
.
time
()
for
i
,
m
in
enumerate
(
metrics
):
metric_list
[
i
].
update
(
np
.
mean
(
m
),
len
(
batch
[
0
])
)
metric_list
[
i
].
update
(
np
.
mean
(
m
),
batch_size
)
fetchs_str
=
''
.
join
([
str
(
m
.
value
)
+
' '
for
m
in
metric_list
]
+
[
batch_time
.
value
])
+
's'
ips_info
=
" ips: {:.5f} images/sec."
.
format
(
batch_size
/
batch_time
.
val
)
fetchs_str
+=
ips_info
if
vdl_writer
:
global
total_step
logger
.
scaler
(
'loss'
,
metrics
[
0
][
0
],
total_step
,
vdl_writer
)
...
...
@@ -502,37 +508,26 @@ def run(dataloader,
else
:
epoch_str
=
"epoch:{:<3d}"
.
format
(
epoch
)
step_str
=
"{:s} step:{:<4d}"
.
format
(
mode
,
idx
)
# Keep the first 10 batches statistics, They are important for develop
if
epoch
==
0
and
idx
<
10
:
logger
.
info
(
"{:s} {:s} {:s}"
.
format
(
logger
.
coloring
(
epoch_str
,
"HEADER"
)
if
idx
==
0
else
epoch_str
,
logger
.
coloring
(
step_str
,
"PURPLE"
),
logger
.
coloring
(
fetchs_str
,
'OKGREEN'
)))
else
:
if
idx
%
config
.
get
(
'print_interval'
,
10
)
==
0
:
logger
.
info
(
"{:s} {:s} {:s}"
.
format
(
logger
.
coloring
(
epoch_str
,
"HEADER"
)
if
idx
==
0
else
epoch_str
,
logger
.
coloring
(
step_str
,
"PURPLE"
),
logger
.
coloring
(
fetchs_str
,
'OKGREEN'
)))
if
idx
%
config
.
get
(
'print_interval'
,
10
)
==
0
:
logger
.
info
(
"{:s} {:s} {:s}"
.
format
(
epoch_str
if
idx
==
0
else
epoch_str
,
step_str
,
fetchs_str
))
tic
=
time
.
time
()
if
config
.
get
(
'use_dali'
):
dataloader
.
reset
()
end_str
=
''
.
join
([
str
(
m
.
mean
)
+
' '
for
m
in
metric_list
]
+
[
batch_time
.
total
])
+
's'
ips_info
=
"ips: {:.5f} images/sec."
.
format
(
batch_size
*
batch_time
.
count
/
batch_time
.
sum
)
if
mode
==
'eval'
:
logger
.
info
(
"END {:s} {:s}s
"
.
format
(
mode
,
end_str
))
logger
.
info
(
"END {:s} {:s}s
{:s}"
.
format
(
mode
,
end_str
,
ips_info
))
else
:
end_epoch_str
=
"END epoch:{:<3d}"
.
format
(
epoch
)
logger
.
info
(
"{:s} {:s} {:s}"
.
format
(
logger
.
coloring
(
end_epoch_str
,
"RED"
),
logger
.
coloring
(
mode
,
"PURPLE"
),
logger
.
coloring
(
end_str
,
"OKGREEN"
)))
logger
.
info
(
"{:s} {:s} {:s} {:s}"
.
format
(
end_epoch_str
,
mode
,
end_str
,
ips_info
))
# return top1_acc in order to save the best model
if
mode
==
'valid'
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录