提交 df4f8b00 编写于 作者: W wuzewu

create parameter when load module

上级 e209d195
......@@ -84,6 +84,25 @@ class Module(object):
self.config = ModuleConfig(self.module_dir)
self.config.load()
self._process_parameter()
def _process_parameter(self):
global_block = self.inference_program.global_block()
for param in self.config.desc.parameters:
name = param.name
if name in global_block.vars:
var = global_block.vars[name]
global_block.create_parameter(
name=name,
trainable=param.trainable,
shape=var.shape,
dtype=var.dtype,
optimize_attr={'learning_rate': param.learning_rate},
type=var.type,
lod_level=var.lod_level,
error_clip=var.error_clip,
stop_gradient=var.stop_gradient,
is_data=var.is_data)
def _construct_feed_dict(self, inputs):
""" Construct feed dict according to user's inputs and module config.
......@@ -253,8 +272,9 @@ class ModuleConfig(object):
with open(pb_path, "rb") as fi:
self.desc.ParseFromString(fi.read())
print("self.desc.sign2var",
self.desc.sign2var["default"].feed_desc[0].var_name)
# print("self.desc.sign2var",
# self.desc.sign2var["default"].feed_desc[0].var_name)
if self.desc.contain_assets:
# load assets
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册