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

Export ALBERT main layer in TensorFlow (#3354)

Browse Source
This commit is contained in:
Lysandre Debut
2020-03-19 13:53:05 -04:00
committed by GitHub
parent 3bedfd3347
commit f049be7ad4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/__init__.py
View File

@@ -434,6 +434,7 @@ if is_tf_available():
from .modeling_tf_albert import ( from .modeling_tf_albert import (
TFAlbertPreTrainedModel, TFAlbertPreTrainedModel,
TFAlbertMainLayer,
TFAlbertModel, TFAlbertModel,
TFAlbertForMaskedLM, TFAlbertForMaskedLM,
TFAlbertForSequenceClassification, TFAlbertForSequenceClassification,