Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
955f7660
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
大约 1 年 前同步成功
通知
282
Star
12117
Fork
2091
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
200
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleHub
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
200
Issue
200
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
955f7660
编写于
3月 10, 2020
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'release/v1.5' into release/v1.6
上级
a17fd08b
81a8fcd9
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
19 addition
and
1 deletion
+19
-1
RELEASE.md
RELEASE.md
+12
-0
paddlehub/finetune/task/base_task.py
paddlehub/finetune/task/base_task.py
+7
-1
未找到文件。
RELEASE.md
浏览文件 @
955f7660
# `v1.5.4`
*
修复Fine-tune中断,checkpoint文件恢复训练失败的问题
# `v1.5.3`
*
优化口罩模型输出结果,提供更加灵活的部署及调用方式
# `v1.5.2`
*
优化pyramidbox_lite_server_mask、pyramidbox_lite_mobile_mask模型的服务化部署性能
# `v1.5.1`
# `v1.5.1`
*
修复加载module缺少cache目录的问题
*
修复加载module缺少cache目录的问题
...
...
paddlehub/finetune/task/base_task.py
浏览文件 @
955f7660
...
@@ -807,10 +807,16 @@ class BaseTask(object):
...
@@ -807,10 +807,16 @@ class BaseTask(object):
# NOTE: current saved checkpoint machanism is not completed,
# NOTE: current saved checkpoint machanism is not completed,
# it can't restore dataset training status
# it can't restore dataset training status
def
save_checkpoint
(
self
):
def
save_checkpoint
(
self
):
"""
save the program of the last step in training
"""
model_saved_dir
=
os
.
path
.
join
(
self
.
config
.
checkpoint_dir
,
model_saved_dir
=
os
.
path
.
join
(
self
.
config
.
checkpoint_dir
,
"step_%d"
%
self
.
current_step
)
"step_%d"
%
self
.
current_step
)
logger
.
info
(
"Saving model checkpoint to {}"
.
format
(
model_saved_dir
))
logger
.
info
(
"Saving model checkpoint to {}"
.
format
(
model_saved_dir
))
self
.
save_inference_model
(
dirname
=
model_saved_dir
)
# to resume traning by loading ckpt, it must be save program (save_persistables)
fluid
.
io
.
save_persistables
(
self
.
exe
,
dirname
=
model_saved_dir
,
main_program
=
self
.
main_program
)
save_checkpoint
(
save_checkpoint
(
checkpoint_dir
=
self
.
config
.
checkpoint_dir
,
checkpoint_dir
=
self
.
config
.
checkpoint_dir
,
current_epoch
=
self
.
current_epoch
,
current_epoch
=
self
.
current_epoch
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录