kaggle google quest:Google QUEST Q&A标签Kaggle竞赛第六名解决方案 源码

kaggle-google-quest 在Google QUEST Q&A Labeling Kaggle竞赛中训练第六名解决方案的变压器模型的代码。 详细说明发布在的Kaggle讨论部分 运行代码需要: 比赛数据放在名为data的目录中。 可以从下载。 需要安装requirements.txt文件中的软件包。 默认情况下,训练是在GPU(单个RTX 2080Ti)上进行的,因此需要CUDA以及大约10GB的GPU内存。 为了解决CUDA内存不足的错误,可以在train.py脚本中将批大小减小到1,并将梯度train.py增加到8。 要重现所有4种变压器模型,请运行以下命令: python train.py -model_name=siamese_roberta && python finetune.py -model_name=siamese_roberta python