From 64cde5d16de78181e348abc85fa45cee29512ec6 Mon Sep 17 00:00:00 2001 From: LiuHao Date: Thu, 6 Aug 2020 18:23:44 +0800 Subject: [PATCH] Update run_ernie_classifier.py (#4790) --- .../sentiment_classification/run_ernie_classifier.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/PaddleNLP/sentiment_classification/run_ernie_classifier.py b/PaddleNLP/sentiment_classification/run_ernie_classifier.py index 21ab7426..f667aa68 100644 --- a/PaddleNLP/sentiment_classification/run_ernie_classifier.py +++ b/PaddleNLP/sentiment_classification/run_ernie_classifier.py @@ -38,16 +38,16 @@ from utils import init_checkpoint def ernie_pyreader(args, pyreader_name): - src_ids = fluid.data( + src_ids = fluid.layers.data( name="src_ids", shape=[None, args.max_seq_len, 1], dtype="int64") - sent_ids = fluid.data( + sent_ids = fluid.layers.data( name="sent_ids", shape=[None, args.max_seq_len, 1], dtype="int64") - pos_ids = fluid.data( + pos_ids = fluid.layers.data( name="pos_ids", shape=[None, args.max_seq_len, 1], dtype="int64") - input_mask = fluid.data( + input_mask = fluid.layers.data( name="input_mask", shape=[None, args.max_seq_len, 1], dtype="float32") - labels = fluid.data(name="labels", shape=[None, 1], dtype="int64") - seq_lens = fluid.data(name="seq_lens", shape=[None], dtype="int64") + labels = fluid.layers.data(name="labels", shape=[None, 1], dtype="int64") + seq_lens = fluid.layers.data(name="seq_lens", shape=[None], dtype="int64") pyreader = fluid.io.DataLoader.from_generator( feed_list=[src_ids, sent_ids, pos_ids, input_mask, labels, seq_lens], -- GitLab