fix: arg_opt is not defined

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")

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")