SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

AlbertForSequenceClassification

Browse Source
This commit is contained in:
Lysandre
2019-11-01 15:17:31 +00:00
committed by Lysandre Debut
parent 0d07a23c04
commit 6637a77f80
2 changed files with 77 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
transformers/__init__.py
View File

@@ -107,7 +107,8 @@ if is_torch_available():
CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model
from .modeling_albert import (AlbertModel, AlbertForMaskedLM, load_tf_weights_in_albert, ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
from .modeling_albert import (AlbertModel, AlbertForMaskedLM, AlbertForSequenceClassification,
load_tf_weights_in_albert, ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
# Optimization
from .optimization import (AdamW, get_constant_schedule, get_constant_schedule_with_warmup, get_cosine_schedule_with_warmup,