From 314fef8ecc9e305716cc371ce4fcaf8a8b32b2ef Mon Sep 17 00:00:00 2001 From: jonyguo Date: Tue, 1 Sep 2020 10:08:32 +0800 Subject: [PATCH] fix: arg_opt is not defined --- model_zoo/official/nlp/tinybert/run_general_distill.py | 2 +- model_zoo/official/nlp/tinybert/run_task_distill.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/model_zoo/official/nlp/tinybert/run_general_distill.py b/model_zoo/official/nlp/tinybert/run_general_distill.py index 36b88487df..1b64ed4dfb 100644 --- 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") diff --git a/model_zoo/official/nlp/tinybert/run_task_distill.py b/model_zoo/official/nlp/tinybert/run_task_distill.py index e0e78c4ed0..0b93a2b3bd 100644 --- 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")