EADST

Bert-Small Pytorch Model

Bert-Small Pytorch Model

from transformers import AutoModel
from transformers import AutoConfig

model_name = "google/bert_uncased_L-4_H-512_A-8"
config = AutoConfig.from_pretrained(model_name)
self.model = AutoModel.from_pretrained(model_name, add_pooling_layer=False, config=config)

Reference:

google/bert_uncased_L-4_H-512_A-8

相关标签
About Me
XD
Goals determine what you are going to be.
Category
标签云
站点统计

本站现有博文242篇,共被浏览290900

本站已经建立1780天!

热门文章
文章归档
回到顶部