From f08a05975e27b9536e6c20a1514af99e24e1219e Mon Sep 17 00:00:00 2001 From: jonyguo Date: Tue, 1 Sep 2020 17:28:09 +0800 Subject: [PATCH] fix: data_tpye var error --- model_zoo/official/nlp/tinybert/run_task_distill.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/nlp/tinybert/run_task_distill.py b/model_zoo/official/nlp/tinybert/run_task_distill.py index 0b93a2b3bd..275207ca9e 100644 --- a/model_zoo/official/nlp/tinybert/run_task_distill.py +++ b/model_zoo/official/nlp/tinybert/run_task_distill.py @@ -134,7 +134,7 @@ def run_predistill(): dataset = create_tinybert_dataset('td', td_teacher_net_cfg.batch_size, device_num, rank, args_opt.do_shuffle, args_opt.train_data_dir, args_opt.schema_dir, - data_tpye=dataset_type) + data_type=dataset_type) dataset_size = dataset.get_dataset_size() print('td1 dataset size: ', dataset_size)