Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
f6fffd8c
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看板
提交
f6fffd8c
编写于
11月 10, 2020
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix model compatibility issues
上级
601c81d2
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
19 addition
and
8 deletion
+19
-8
paddlehub/compat/module/module_v1.py
paddlehub/compat/module/module_v1.py
+6
-1
paddlehub/compat/task/base_task.py
paddlehub/compat/task/base_task.py
+13
-7
未找到文件。
paddlehub/compat/module/module_v1.py
浏览文件 @
f6fffd8c
...
...
@@ -211,7 +211,12 @@ class ModuleV1(object):
'''Load the infomation of Module object defined in the specified directory.'''
desc_file
=
os
.
path
.
join
(
directory
,
'module_desc.pb'
)
desc
=
module_v1_utils
.
convert_module_desc
(
desc_file
)
return
desc
.
module_info
# The naming of some old versions of Module is not standardized, which format of uppercase
# letters. This will cause the path of these modules to be incorrect after installation.
module_info
=
desc
.
module_info
module_info
.
name
=
module_info
.
name
.
lower
()
return
module_info
def
assets_path
(
self
):
return
os
.
path
.
join
(
self
.
directory
,
'assets'
)
...
...
paddlehub/compat/task/base_task.py
浏览文件 @
f6fffd8c
...
...
@@ -117,6 +117,8 @@ class BaseTask(object):
self
.
_base_data_reader
=
data_reader
self
.
_base_feed_list
=
feed_list
self
.
_compatible_mode
=
True
if
data_reader
else
False
@
contextlib
.
contextmanager
def
phase_guard
(
self
,
phase
:
str
):
self
.
enter_phase
(
phase
)
...
...
@@ -308,6 +310,10 @@ class BaseTask(object):
return
wrapper
if
self
.
_compatible_mode
:
self
.
env
.
generator
=
self
.
_base_data_reader
.
data_generator
(
batch_size
=
self
.
config
.
batch_size
,
phase
=
self
.
phase
,
data
=
self
.
_predict_data
,
return_list
=
True
)
else
:
if
self
.
is_predict_phase
:
records
=
self
.
_predict_data
else
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录