Onion12138 / CasRelPyTorch

Reimplement CasRel model in PyTorch.使用PyTorch对吉林大学CasRel模型进行复现,并在百度关系抽取数据集上训练测试。
184 stars 26 forks source link

测试时的batch_size #7

Open 1qingqing1 opened 2 years ago

1qingqing1 commented 2 years ago

请问测试时是只能一条条数据进行测试,无法进行批量处理吗,那这样的话,用cpu和gpu是不是推理速度差不多呢

BraveRab commented 2 years ago

请问测试时是只能一条条数据进行测试,无法进行批量处理吗,那这样的话,用cpu和gpu是不是推理速度差不多呢

请问,我也是result.json文件输出只有一条数据,想请教一下这一条数据是哪个数据集下面的呀?不能整体输出预测结果吗?