Skip to content

The repo provides a basic training framework built from Huggingface Transformers

License

Notifications You must be signed in to change notification settings

panzeyu2013/trans-trainer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is a repo to save my training framework

Modified from mPLUG-Owl2 and Transformers repo source code

The first edition takes an example for video/image captioning tasks and trying to be more general for other tasks

pyproject.toml -> add fvcore [WIP]

train/custom_model_trainner.py -> add support for list of labels as input to evaluate results. [Done]

scripts [Done]

Test after training [Done]

First implemention [WIP]

About

The repo provides a basic training framework built from Huggingface Transformers

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published