提交 fda184fd 编写于 作者: Z Zeyu Chen

remove useless python import

上级 7a1c7a7d
...@@ -17,9 +17,6 @@ from __future__ import print_function ...@@ -17,9 +17,6 @@ from __future__ import print_function
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from urllib.request import urlretrieve
from tqdm import tqdm
import os import os
import sys import sys
import hashlib import hashlib
......
# coding: utf-8
# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved. # Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
# #
# Licensed under the Apache License, Version 2.0 (the "License" # Licensed under the Apache License, Version 2.0 (the "License"
...@@ -163,6 +162,8 @@ def train(use_cuda=False): ...@@ -163,6 +162,8 @@ def train(use_cuda=False):
target_vars=[predict_word], target_vars=[predict_word],
executor=exe) executor=exe)
create
dictionary = defaultdict(int) dictionary = defaultdict(int)
w_id = 0 w_id = 0
for w in word_dict: for w in word_dict:
...@@ -218,13 +219,17 @@ def test_save_module(use_cuda=False): ...@@ -218,13 +219,17 @@ def test_save_module(use_cuda=False):
dictionary[w] = w_id dictionary[w] = w_id
w_id += 1 w_id += 1
input_desc = {"words": words.name} signature = hub.create_singature(
output_desc = {"emb": word_emb.name} "default", inputs=[data], outputs=[word_emb])
config = hub.ModuleConfig(saved_module_dir) hub.create_module(
config.register_feed_signature(input_desc, sign_name="default") sign_arr=signature, program=main_program, path=save_module_dir)
config.register_fetch_signature(output_desc, sign_name="default") # input_desc = {"words": words.name}
config.save_dict(word_dict=dictionary) # output_desc = {"emb": word_emb.name}
config.dump() # config = hub.ModuleConfig(saved_module_dir)
# config.register_feed_signature(input_desc, sign_name="default")
# config.register_fetch_signature(output_desc, sign_name="default")
# config.save_dict(word_dict=dictionary)
# config.dump()
def test_load_module(use_cuda=False): def test_load_module(use_cuda=False):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册