Skip to content

GitLab

Commit f7115752 authored by Martin Malmsten's avatar Martin Malmsten
Browse files

Added test for AlbertForTokenClassification

parent aceb6a09
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_albert.py
View file @ f7115752
...@@ -29,6 +29,7 @@ if is_torch_available(): ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment