Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PLSC
提交
59104538
P
PLSC
项目概览
PaddlePaddle
/
PLSC
通知
10
Star
3
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
5
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PLSC
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
5
Issue
5
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
59104538
编写于
12月 23, 2019
作者:
L
lilong12
提交者:
GitHub
12月 23, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
remove input func (#13)
上级
fbb20a31
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
6 addition
and
16 deletion
+6
-16
plsc/entry.py
plsc/entry.py
+6
-16
未找到文件。
plsc/entry.py
浏览文件 @
59104538
...
@@ -450,16 +450,10 @@ class Entry(object):
...
@@ -450,16 +450,10 @@ class Entry(object):
checkpoint_dir
=
self
.
checkpoint_dir
checkpoint_dir
=
self
.
checkpoint_dir
if
self
.
fs_name
is
not
None
:
if
self
.
fs_name
is
not
None
:
ans
=
'y'
if
os
.
path
.
exists
(
checkpoint_dir
):
if
os
.
path
.
exists
(
checkpoint_dir
):
ans
=
input
(
"Downloading pretrained models, but the local "
logger
.
info
(
"Local dir {} exists, we'll overwrite it."
.
format
(
"checkpoint directory ({}) exists, overwrite it "
checkpoint_dir
))
"or not? [Y/N]"
.
format
(
checkpoint_dir
))
shutil
.
rmtree
(
checkpoint_dir
)
if
ans
.
lower
()
==
'y'
:
if
os
.
path
.
exists
(
checkpoint_dir
):
logger
.
info
(
"Using the local checkpoint directory."
)
shutil
.
rmtree
(
checkpoint_dir
)
os
.
makedirs
(
checkpoint_dir
)
os
.
makedirs
(
checkpoint_dir
)
# sync all trainers to avoid loading checkpoints before
# sync all trainers to avoid loading checkpoints before
...
@@ -537,12 +531,8 @@ class Entry(object):
...
@@ -537,12 +531,8 @@ class Entry(object):
assert
self
.
model_save_dir
,
\
assert
self
.
model_save_dir
,
\
"Does not set model_save_dir for inference model converting."
"Does not set model_save_dir for inference model converting."
if
os
.
path
.
exists
(
self
.
model_save_dir
):
if
os
.
path
.
exists
(
self
.
model_save_dir
):
ans
=
input
(
"model_save_dir for inference model ({}) exists, "
logger
.
info
(
"model_save_dir for inference model ({}) exists, "
"overwrite it or not? [Y/N]"
.
format
(
model_save_dir
))
"we will overwrite it."
.
format
(
self
.
model_save_dir
))
if
ans
.
lower
()
==
'n'
:
logger
.
error
(
"model_save_dir for inference model exists, "
"and cannot overwrite it."
)
exit
()
shutil
.
rmtree
(
self
.
model_save_dir
)
shutil
.
rmtree
(
self
.
model_save_dir
)
fluid
.
io
.
save_inference_model
(
self
.
model_save_dir
,
fluid
.
io
.
save_inference_model
(
self
.
model_save_dir
,
feeded_var_names
=
[
image
.
name
],
feeded_var_names
=
[
image
.
name
],
...
@@ -550,7 +540,7 @@ class Entry(object):
...
@@ -550,7 +540,7 @@ class Entry(object):
executor
=
exe
,
executor
=
exe
,
main_program
=
main_program
)
main_program
=
main_program
)
if
self
.
fs_name
:
if
self
.
fs_name
:
self
.
put_files_to_hdfs
(
model_save_dir
)
self
.
put_files_to_hdfs
(
self
.
model_save_dir
)
def
predict
(
self
):
def
predict
(
self
):
model_name
=
self
.
model_name
model_name
=
self
.
model_name
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录