sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!5568 bert script bugfix 

Merge pull request !5568 from yoonlee666/master
This commit is contained in:
mindspore-ci-bot 2020-08-31 19:44:21 +08:00 committed by Gitee
parent 7371ceddde ff4e02a09d
commit 042ac51f05
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
model_zoo/official/nlp/bert/run_classifier.py
View File

@ -136,9 +136,10 @@ def run_classifier():
parser = argparse.ArgumentParser(description="run classifier") parser = argparse.ArgumentParser(description="run classifier")
parser.add_argument("--device_target", type=str, default="Ascend", choices=["Ascend", "GPU"], parser.add_argument("--device_target", type=str, default="Ascend", choices=["Ascend", "GPU"],
help="Device type, default is Ascend") help="Device type, default is Ascend")
parser.add_argument("--assessment_method", type=str, default="accuracy", parser.add_argument("--assessment_method", type=str, default="Accuracy",
choices=["Mcc", "Spearman_correlation", "Accuracy"], choices=["Mcc", "Spearman_correlation", "Accuracy", "F1"],
help="assessment_method including [MCC, Spearman_correlation, Accuracy], default is accuracy") help="assessment_method including [Mcc, Spearman_correlation, Accuracy, F1],\
default is Accuracy")
parser.add_argument("--do_train", type=str, default="false", choices=["true", "false"], parser.add_argument("--do_train", type=str, default="false", choices=["true", "false"],
help="Enable train, default is false") help="Enable train, default is false")
parser.add_argument("--do_eval", type=str, default="false", choices=["true", "false"], parser.add_argument("--do_eval", type=str, default="false", choices=["true", "false"],

2
model_zoo/official/nlp/bert/run_ner.py
View File

@ -145,7 +145,7 @@ def run_ner():
parser = argparse.ArgumentParser(description="run classifier") parser = argparse.ArgumentParser(description="run classifier")
parser.add_argument("--device_target", type=str, default="Ascend", choices=["Ascend", "GPU"], parser.add_argument("--device_target", type=str, default="Ascend", choices=["Ascend", "GPU"],
help="Device type, default is Ascend") help="Device type, default is Ascend")
parser.add_argument("--assessment_method", type=str, default="accuracy", choices=["f1", "clue_benchmark"], parser.add_argument("--assessment_method", type=str, default="F1", choices=["F1", "clue_benchmark"],
help="assessment_method include: [F1, clue_benchmark], default is F1") help="assessment_method include: [F1, clue_benchmark], default is F1")
parser.add_argument("--do_train", type=str, default="false", choices=["true", "false"], parser.add_argument("--do_train", type=str, default="false", choices=["true", "false"],
help="Eable train, default is false") help="Eable train, default is false")