Browse Source

fix: data_tpye var error

tags/v1.0.0
jonyguo 5 years ago
parent
commit
f08a05975e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/nlp/tinybert/run_task_distill.py

+ 1
- 1
model_zoo/official/nlp/tinybert/run_task_distill.py View File

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


Loading…
Cancel
Save