Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
2cf8014d
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
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看板
提交
2cf8014d
编写于
4月 27, 2020
作者:
littletomatodonkey
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
save best model
上级
07796136
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
26 addition
and
9 deletion
+26
-9
tools/program.py
tools/program.py
+11
-7
tools/train.py
tools/train.py
+15
-2
未找到文件。
tools/program.py
浏览文件 @
2cf8014d
...
...
@@ -383,17 +383,21 @@ def run(dataloader, exe, program, fetchs, epoch=0, mode='train'):
tic
=
time
.
time
()
for
i
,
m
in
enumerate
(
metrics
):
metric_list
[
i
].
update
(
m
[
0
],
len
(
batch
[
0
]))
fetchs_str
=
''
.
join
([
str
(
m
.
value
)
+
' '
for
m
in
metric_list
]
+
[
batch_time
.
value
])
fetchs_str
=
''
.
join
([
str
(
m
.
value
)
+
' '
for
m
in
metric_list
]
+
[
batch_time
.
value
])
if
epoch
!=
-
1
:
logger
.
info
(
"epoch:{:<3d} {:s} step:{:<4d} {:s}s"
.
format
(
epoch
,
mode
,
idx
,
fetchs_str
))
epoch
,
mode
,
idx
,
fetchs_str
))
else
:
logger
.
info
(
"{:s} step:{:<4d} {:s}s"
.
format
(
mode
,
idx
,
fetchs_str
))
logger
.
info
(
"{:s} step:{:<4d} {:s}s"
.
format
(
mode
,
idx
,
fetchs_str
))
end_str
=
''
.
join
([
str
(
m
.
mean
)
+
' '
for
m
in
metric_list
]
+
[
batch_time
.
total
])
if
epoch
!=
-
1
:
end_str
=
''
.
join
([
str
(
m
.
mean
)
+
' '
for
m
in
metric_list
]
+
[
batch_time
.
total
])
if
epoch
!=
-
1
:
logger
.
info
(
"END epoch:{:<3d} {:s} {:s}s"
.
format
(
epoch
,
mode
,
end_str
))
else
:
logger
.
info
(
"END {:s} {:s}s"
.
format
(
mode
,
end_str
))
# save the best model
top1_acc
=
fetchs
[
"top1"
][
1
].
avg
return
top1_acc
tools/train.py
浏览文件 @
2cf8014d
...
...
@@ -26,6 +26,7 @@ from paddle.fluid.incubate.fleet.collective import fleet
from
ppcls.data
import
Reader
from
ppcls.utils.config
import
get_config
from
ppcls.utils.save_load
import
init_model
,
save_model
from
ppcls.utils
import
logger
import
program
...
...
@@ -61,6 +62,10 @@ def main(args):
startup_prog
=
fluid
.
Program
()
train_prog
=
fluid
.
Program
()
# best_top1_acc_list[0]: top1 acc
# best_top1_acc_list[1]: epoch id
best_top1_acc_list
=
[
0.0
,
0
]
train_dataloader
,
train_fetchs
=
program
.
build
(
config
,
train_prog
,
startup_prog
,
is_train
=
True
)
...
...
@@ -94,8 +99,16 @@ def main(args):
epoch_id
,
'train'
)
# 2. validate with validate dataset
if
config
.
validate
and
epoch_id
%
config
.
valid_interval
==
0
:
program
.
run
(
valid_dataloader
,
exe
,
compiled_valid_prog
,
valid_fetchs
,
epoch_id
,
'valid'
)
top1_acc
=
program
.
run
(
valid_dataloader
,
exe
,
compiled_valid_prog
,
valid_fetchs
,
epoch_id
,
'valid'
)
if
top1_acc
>
best_top1_acc_list
[
0
]:
best_top1_acc_list
[
0
]
=
top1_acc
best_top1_acc_list
[
1
]
=
epoch_id
logger
.
info
(
"Best top1 acc: {}, in epoch: {}"
.
format
(
best_top1_acc_list
[
0
],
best_top1_acc_list
[
1
]))
model_path
=
os
.
path
.
join
(
config
.
model_save_dir
,
config
.
ARCHITECTURE
[
"name"
])
save_model
(
train_prog
,
model_path
,
"best_model"
)
# 3. save the persistable model
if
epoch_id
%
config
.
save_interval
==
0
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录