前面课程当中,我们已经定义好了模型结构,接下来,就可以进行模型训练了。在之前的两个项目里面,我们训练数据比较少,在训练过程中,就只打印了损失值,没有做模型验证。

按照业界的标准流程,应该边训练边验证,以观察模型的训练过程。刚好这份数据集比较全,有验证数据,我们就在这个项目里,走一下边训练边验证的标准流程。

代码示例

1、新建训练文件

内容不可见,请联系管理员开通权限。

2、新建训练文件

内容不可见,请联系管理员开通权限。

3、内存溢出问题

a)调整 batch_size 大小

b)固定预训练模型参数,只训练下游任务

内容不可见,请联系管理员开通权限。

本文为 陈华 原创,欢迎转载,但请注明出处:http://www.ichenhua.cn/edu/note/506