提交 80c3b3f0 编写于 作者: Z Zeyu Chen

remove signature name for load ernie module

上级 5fb669bd
...@@ -68,7 +68,7 @@ if __name__ == '__main__': ...@@ -68,7 +68,7 @@ if __name__ == '__main__':
num_labels = len(reader.get_labels()) num_labels = len(reader.get_labels())
input_dict, output_dict, program = module.context( input_dict, output_dict, program = module.context(
sign_name="tokens", trainable=True, max_seq_len=args.max_seq_len) trainable=True, max_seq_len=args.max_seq_len)
with fluid.program_guard(program): with fluid.program_guard(program):
label = fluid.layers.data(name="label", shape=[1], dtype='int64') label = fluid.layers.data(name="label", shape=[1], dtype='int64')
......
...@@ -68,7 +68,7 @@ if __name__ == '__main__': ...@@ -68,7 +68,7 @@ if __name__ == '__main__':
num_labels = len(reader.get_labels()) num_labels = len(reader.get_labels())
input_dict, output_dict, program = module.context( input_dict, output_dict, program = module.context(
sign_name="tokens", trainable=True, max_seq_len=args.max_seq_len) trainable=True, max_seq_len=args.max_seq_len)
with fluid.program_guard(program): with fluid.program_guard(program):
label = fluid.layers.data(name="label", shape=[1], dtype='int64') label = fluid.layers.data(name="label", shape=[1], dtype='int64')
......
...@@ -67,7 +67,7 @@ if __name__ == '__main__': ...@@ -67,7 +67,7 @@ if __name__ == '__main__':
num_labels = len(reader.get_labels()) num_labels = len(reader.get_labels())
input_dict, output_dict, program = module.context( input_dict, output_dict, program = module.context(
sign_name="tokens", trainable=True, max_seq_len=args.max_seq_len) trainable=True, max_seq_len=args.max_seq_len)
with fluid.program_guard(program): with fluid.program_guard(program):
label = fluid.layers.data( label = fluid.layers.data(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册