严格意义上讲 transformers 并不是 PyTorch 的一部分,然而 transformers 与 PyTorch 或 TensorFlow 结合的太紧密了,而且可以把 transformers 看成是 PyTorch 或 TensorFlow 的延伸,所以也在这里一并讨论了。
transformers 内置了 17 种以 transformer 结构为基础的神经网络:
T5 model
DistilBERT model
ALBERT model
CamemBERT model
XLM-RoBERTa model
Longformer...