cs336Assignment1 训练组件 - 凪风sama

Wait 5 sec.

【摘要】模型训练 对于一个神经网络模型,其训练流程基本如上图所示,本笔记旨在在从零搭建Transformer-CS336作业所搭建的Transformer模型基础上,搭建模型训练所需要的一些基本模块,并最后使用这些模块实现对上述模型的训练与推理验证。依旧感谢【斯坦福 CS336】零基础手搓大模型:分词器、t 阅读全文