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

update typo

上级 72e1fb1f
...@@ -54,7 +54,7 @@ if __name__ == '__main__': ...@@ -54,7 +54,7 @@ if __name__ == '__main__':
batch_size=args.batch_size, batch_size=args.batch_size,
strategy=strategy) strategy=strategy)
# loading Paddlehub BERT # loading Paddlehub ERNIE pretrained model
module = hub.Module(name="ernie") module = hub.Module(name="ernie")
# Sentence classification dataset reader # Sentence classification dataset reader
...@@ -76,7 +76,7 @@ if __name__ == '__main__': ...@@ -76,7 +76,7 @@ if __name__ == '__main__':
pooled_output = output_dict["pooled_output"] pooled_output = output_dict["pooled_output"]
# Setup feed list for data feeder # 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 = [ feed_list = [
input_dict["input_ids"].name, input_dict["position_ids"].name, input_dict["input_ids"].name, input_dict["position_ids"].name,
input_dict["segment_ids"].name, input_dict["input_mask"].name, input_dict["segment_ids"].name, input_dict["input_mask"].name,
......
...@@ -55,7 +55,7 @@ if __name__ == '__main__': ...@@ -55,7 +55,7 @@ if __name__ == '__main__':
batch_size=args.batch_size, batch_size=args.batch_size,
strategy=strategy) strategy=strategy)
# loading Paddlehub ERNIE # loading Paddlehub ERNIE pretrained model
module = hub.Module(name="ernie") module = hub.Module(name="ernie")
# Sequence Label dataset reader # Sequence Label dataset reader
...@@ -79,7 +79,7 @@ if __name__ == '__main__': ...@@ -79,7 +79,7 @@ if __name__ == '__main__':
sequence_output = output_dict["sequence_output"] sequence_output = output_dict["sequence_output"]
# Setup feed list for data feeder # 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 = [ feed_list = [
input_dict["input_ids"].name, input_dict["position_ids"].name, input_dict["input_ids"].name, input_dict["position_ids"].name,
input_dict["segment_ids"].name, input_dict["input_mask"].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.
先完成此消息的编辑!
想要评论请 注册