提交 f08a0597 编写于 作者: J jonyguo

fix: data_tpye var error

上级 f42b3bbf
......@@ -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.
先完成此消息的编辑!
想要评论请 注册