Skip to content

一个简易的nlp模型集合。实现部分经典的nlp模型

Notifications You must be signed in to change notification settings

zouweidong91/nlpcol

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🗂️ nlpcol

nlpcol(nlp collction): 一个简单的nlp基础模型集合。

相较于transformers,本项目尽量以较低的代码实现一部分经典模型。旨在通过构建简单易懂的代码,让初学者能够快速接触及掌握各经典模型以及训练推理过程的各个细节。当前已集成模型: bert, unilm, gpt, mt5, gpt2。

💡 TODO

  • 训练逻辑优化:混合精度、grad_checkoutpoint
  • 训练日志优化
  • 常用优化器实现

🚀 预训练权重

所有权重均为torch版本

模型分类 模型名称 权重来源 官方项目地址
bert bert-base-chinese 谷歌中文bert bert
mt5 mt5-base 谷歌多语言版T5 t5
gpt openai-gpt openai-gpt1 finetune-transformer-lm
gpt CDial-GPT_LCCC-base 清华coai CDial-GPT
gpt2 openai-gpt2 openai-gpt2 gpt-2

ref:

https://github.com/huggingface/transformers
https://github.com/bojone/bert4keras
https://github.com/Tongjilibo/bert4torch

About

一个简易的nlp模型集合。实现部分经典的nlp模型

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages