Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
277aae4a
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看板
提交
277aae4a
编写于
1月 07, 2019
作者:
Z
Zeyu Chen
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix creator mkdir issues and add all test for travis-ci
上级
8c83f793
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
8 addition
and
6 deletion
+8
-6
paddle_hub/module_creator.py
paddle_hub/module_creator.py
+3
-3
tests/tclist_all
tests/tclist_all
+3
-0
tests/test_train_w2v.py
tests/test_train_w2v.py
+2
-3
未找到文件。
paddle_hub/module_creator.py
浏览文件 @
277aae4a
...
...
@@ -20,6 +20,7 @@ import paddle_hub.module_desc_pb2 as modulepb
import
paddle.fluid
as
fluid
from
paddle_hub.utils
import
to_list
from
paddle_hub.signature
import
Signature
from
paddle_hub.module
import
mkdir
import
os
...
...
@@ -31,13 +32,12 @@ def create_module(sign_arr, program, path=None, assets=None):
if
not
path
:
path
=
os
.
path
.
join
(
"."
,
"hub_module"
)
assert
not
os
.
path
.
exists
(
path
),
"path %s should not be existed"
%
path
# create module path for saving
mkdir
(
path
)
module
=
modulepb
.
ModuleDesc
()
program
=
program
.
clone
()
os
.
makedirs
(
path
)
# TODO(wuzewu): save assets data
if
not
assets
:
module
.
contain_assets
=
False
...
...
tests/tclist_all
浏览文件 @
277aae4a
test_downloader
test_export_n_load_module
test_module
test_train_w2v
tests/test_train_w2v.py
浏览文件 @
277aae4a
...
...
@@ -92,7 +92,6 @@ def train():
for
epoch
in
range
(
0
,
PASS_NUM
):
for
mini_batch
in
batch_reader
():
# 定义输入变量
feed_var_list
=
[
main_program
.
global_block
().
var
(
"firstw"
),
main_program
.
global_block
().
var
(
"secondw"
),
...
...
@@ -105,9 +104,9 @@ def train():
main_program
,
feed
=
feeder
.
feed
(
mini_batch
),
fetch_list
=
[
avg_cost
])
print
(
"Cost = %f"
%
cost
[
0
]
)
print
(
"Epoch {} Cost = {}"
.
format
(
epoch
,
cost
[
0
])
)
model_dir
=
"./w2v_model"
model_dir
=
"./
tmp/
w2v_model"
var_list_to_saved
=
[
main_program
.
global_block
().
var
(
"embedding"
)]
print
(
"saving model to %s"
%
model_dir
)
fluid
.
io
.
save_vars
(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录