From 80c3b3f061bab0c0e48590c3f4eaa4b9f11bf4fc Mon Sep 17 00:00:00 2001 From: Zeyu Chen Date: Thu, 11 Apr 2019 15:46:12 +0800 Subject: [PATCH] remove signature name for load ernie module --- demo/ernie-classification/question_answering.py | 2 +- demo/ernie-classification/question_matching.py | 2 +- demo/ernie-seq-labeling/finetune_with_hub.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/demo/ernie-classification/question_answering.py b/demo/ernie-classification/question_answering.py index b09ee782..b53b7412 100644 --- a/demo/ernie-classification/question_answering.py +++ b/demo/ernie-classification/question_answering.py @@ -68,7 +68,7 @@ if __name__ == '__main__': num_labels = len(reader.get_labels()) 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): label = fluid.layers.data(name="label", shape=[1], dtype='int64') diff --git a/demo/ernie-classification/question_matching.py b/demo/ernie-classification/question_matching.py index 64ee1f79..c5cc2d71 100644 --- a/demo/ernie-classification/question_matching.py +++ b/demo/ernie-classification/question_matching.py @@ -68,7 +68,7 @@ if __name__ == '__main__': num_labels = len(reader.get_labels()) 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): label = fluid.layers.data(name="label", shape=[1], dtype='int64') diff --git a/demo/ernie-seq-labeling/finetune_with_hub.py b/demo/ernie-seq-labeling/finetune_with_hub.py index 608d0e0f..1424fdbc 100644 --- a/demo/ernie-seq-labeling/finetune_with_hub.py +++ b/demo/ernie-seq-labeling/finetune_with_hub.py @@ -67,7 +67,7 @@ if __name__ == '__main__': num_labels = len(reader.get_labels()) 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): label = fluid.layers.data( -- GitLab