Skip to content

keshavd/Decepticons

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

82 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

 Decepticons

Extending HuggingFace's transformers package with new models.

Description

Many of HuggingFace's finetuning models have redundant code. These patterns have been refactored into Mixins and new generalized training Heads allowing for more flexibility.

New Models

  • BERT

    • BertForSequenceClassification
    • BertForTokenClassification
  • RoBERTA

    • RobertaForSequenceClassification
    • RobertaForTokenClassification

To Be Added:

  • BERT
    • BertForTokenCrfClassification

About

A transformers extension

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages