[组队学习]--第一次训练baseline

本文记录了参加天池-中文预训练模型泛化能力挑战赛的经历,包括从代码下载、环境配置(解决Pytorch与CUDA版本问题)、模型训练(调整batchSize和epochs)到遇到并解决UnboundLocalError。接着,介绍了Docker的安装、镜像构建与推送至阿里云仓库的过程,以及提交时的注意事项,如设置固定密码和文件位置的合规性。
摘要由CSDN通过智能技术生成

训练代码

参加的 天池-中文预训练模型泛化能力挑战赛 ,根据Datawhale提供的代码,在自己的机器上跑一遍训练流程。

准备阶段

代码下载

clone 下来代码,通过Pycharm打开项目。

环境配置

根据readme中的提示,下载 Pytorch,transformers,sklearn。

在之后的代码运行中,遇到了Pytorch版本不匹配的问题。我才开始安装的是 1.1.0,不行后面又安装的1.2.0,也不行。但是我的CUDA版本为10.0,只能支持到1.2.0,所以又把CUDA升级到10.1,安装的Pytorch 1.6.0。

感觉在以后运行课题项目代码的时候又要把CUDA安装到10.0版本(版本问题太坑了-.-)

代码运行

因为我的机器配置太低,所以在开始运行的时候我把 batchSize改成了 8 ,但是还是不行,之后改成了4也不行(显卡内存太少了),最后改成了2 ,可以运行了。
而且为了第一遍熟悉流程,把 epochs

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值