uniex训练时速度很快,但是测试时候速度很慢
miaomiao1215 opened this issue · 1 comments
miaomiao1215 commented
uniex训练时速度很快,但是测试时候速度很慢
miaomiao1215 commented
代码Fengshenbang-LM-main/fengshen/models/uniex/modeling_uniex.py第1265和1266行(如下)应该去掉,否则1279行会报错
if 'tokens' in batch_data[i].keys():
del batch_data[i]['tokens']