提交 8580f98b 编写于 作者: Z Zeyu Chen

update typo

上级 72e1fb1f
......@@ -54,7 +54,7 @@ if __name__ == '__main__':
batch_size=args.batch_size,
strategy=strategy)
# loading Paddlehub BERT
# loading Paddlehub ERNIE pretrained model
module = hub.Module(name="ernie")
# Sentence classification dataset reader
......@@ -76,7 +76,7 @@ if __name__ == '__main__':
pooled_output = output_dict["pooled_output"]
# Setup feed list for data feeder
# Must feed all the tensor of bert's module need
# Must feed all the tensor of ERNIE's module need
feed_list = [
input_dict["input_ids"].name, input_dict["position_ids"].name,
input_dict["segment_ids"].name, input_dict["input_mask"].name,
......
......@@ -55,7 +55,7 @@ if __name__ == '__main__':
batch_size=args.batch_size,
strategy=strategy)
# loading Paddlehub ERNIE
# loading Paddlehub ERNIE pretrained model
module = hub.Module(name="ernie")
# Sequence Label dataset reader
......@@ -79,7 +79,7 @@ if __name__ == '__main__':
sequence_output = output_dict["sequence_output"]
# Setup feed list for data feeder
# Must feed all the tensor of bert's module need
# Must feed all the tensor of ERNIE's module need
feed_list = [
input_dict["input_ids"].name, input_dict["position_ids"].name,
input_dict["segment_ids"].name, input_dict["input_mask"].name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册