[组队学习]--第一次训练baseline
训练代码参加的 天池-中文预训练模型泛化能力挑战赛 ,根据Datawhale提供的代码,在自己的机器上跑一遍训练流程。准备阶段代码下载clone 下来代码,通过Pycharm打开项目。环境配置根据readme中的提示,下载 Pytorch,transformers,sklearn。在之后的代码运行中,遇到了Pytorch版本不匹配的问题。我才开始安装的是 1.1.0,不行后面又安装的1.2.0,也不行。但是我的CUDA版本为10.0,只能支持到1.2.0,所以又把CUDA升级到10.1,安装
原创
2021-02-21 18:37:37 ·
200 阅读 ·
0 评论