Browse Source

!5628 fix: model zoo tinybert err

Merge pull request !5628 from guozhijian/fix_tinybert_err
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
0a203c68bd
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