p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: arg_opt is not defined

This commit is contained in:
jonyguo 2020-09-01 10:08:32 +08:00
parent 79b117fe02
commit 314fef8ecc
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/nlp/tinybert/run_general_distill.py
View File

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

4
model_zoo/official/nlp/tinybert/run_task_distill.py
View File

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