riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Added test for AlbertForTokenClassification

Browse Source
This commit is contained in:
Martin Malmsten 2020-02-27 12:24:20 +01:00
parent aceb6a0907
commit f71157529e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_albert.py
View File

@ -29,6 +29,7 @@ if is_torch_available():
AlbertModel, AlbertModel,
AlbertForMaskedLM, AlbertForMaskedLM,
AlbertForSequenceClassification, AlbertForSequenceClassification,
AlbertForTokenClassification,
AlbertForQuestionAnswering, AlbertForQuestionAnswering,
) )
from transformers.modeling_albert import ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP from transformers.modeling_albert import ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP