Skip to content

A simple framework to pre-train and fine-tune T5 model with pytorch-lightning and transformers

Notifications You must be signed in to change notification settings

TianHongZXY/T5-pytorch-lightning

Repository files navigation

T5 pytorch lightning

基于 pytorch lightning 和 🤗 transformers 预训练和 fine-tune T5 模型

经过测试已支持预训练和fine-tune的模型有:google/t5google/mt5LangBoat/MengziIDEA-CCNL/Randeng-770M,使用Randeng需要将Fengshen框架中的fengshen文件夹放到根目录。

run.slurm 中修改训练参数

测试环境 transformers==4.12.0, pytorch-lightning==1.5.10, torch==1.10.0

About

A simple framework to pre-train and fine-tune T5 model with pytorch-lightning and transformers

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published