提交 0a203c68 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5628 fix: model zoo tinybert err

Merge pull request !5628 from guozhijian/fix_tinybert_err
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册