fix: arg_opt is not defined

5 years ago · 314fef8ecc
parent 79b117fe02
commit 314fef8ecc
2 changed files with 3 additions and 3 deletions
--- a/model_zoo/official/nlp/tinybert/run_general_distill.py
+++ b/model_zoo/official/nlp/tinybert/run_general_distill.py
@ -106,7 +106,7 @@ def run_general_distill():
    if args_opt.dataset_type == "tfrecord":
        dataset_type = DataType.TFRECORD
-    elif arg_opt.dataset_type == "mindrecord":
+    elif args_opt.dataset_type == "mindrecord":
        dataset_type = DataType.MINDRECORD
    else:
        raise Exception("dataset format is not supported yet")
--- a/model_zoo/official/nlp/tinybert/run_task_distill.py
+++ b/model_zoo/official/nlp/tinybert/run_task_distill.py
@ -125,9 +125,9 @@ def run_predistill():
    rank = 0
    device_num = 1
-    if arg_opt.dataset_type == "tfrecord":
+    if args_opt.dataset_type == "tfrecord":
        dataset_type = DataType.TFRECORD
-    elif arg_opt.dataset_type == "mindrecord":
+    elif args_opt.dataset_type == "mindrecord":
        dataset_type = DataType.MINDRECORD
    else:
        raise Exception("dataset format is not supported yet")