README.md 627 Bytes
Newer Older
1
# TensorFlow Natural Language Processing Modelling Toolkit
Hongkun Yu's avatar
Hongkun Yu committed
2

3
4
5
tensorflow/models/official/nlp provides a [modeling library](modeling) for constructing
NLP model achitectures, as well as TF2 reference implementations for
state-of-the-art models.
Hongkun Yu's avatar
Hongkun Yu committed
6

7
8
The repository contains the following models, with implementations, pre-trained
model weights, usage scripts and conversion utilities:
Hongkun Yu's avatar
Hongkun Yu committed
9

Hongkun Yu's avatar
Hongkun Yu committed
10
* [Bert](bert)
11
* [Albert](albert)
Hongkun Yu's avatar
Hongkun Yu committed
12
* [XLNet](xlnet)
Hongkun Yu's avatar
Hongkun Yu committed
13
* [Transformer for translation](transformer)
14
15
16
17
18
19
20
21

Addtional features:

* Distributed trainable on both multi-GPU and TPU
* e2e training for custom models, including both pretraining and finetuning.