【源码阅读】BERT pytorch源码结构关系图

在这里插入图片描述
如图,模型基本结构是BERT,是model,而BERTLM集成了Bert,NSP,MLM,是模型结构的核心。
在数据的处理中,核心是BERTDataset
训练或者评估的时候,调用了BERTTrainer,他使用了BertDataset的数据来对BERTLM进行训练与迭代,

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章