/transformer-heads

Toolkit for attaching, training, saving and loading of new heads for transformer models

Primary LanguageJupyter NotebookMIT LicenseMIT