pytorch transformers ....

pytorch transformers ....

transformers的预训练模型下载本地特定位置默认是在~/.cache/huggingface/transformers

model = GPT2LMHeadModel.from_pretrained('gpt2', cache_dir="...")

想知道transformers的模型都是什么结构的,比如bert模型:

transformers/models/bert/__init__.py

这里可以看到导入了

from .modeling_bert import (            BERT_PRETRAINED_MODEL_ARCHIVE_LIST,            BertForMaskedLM,            BertForMultipleChoice,            BertForNextSentencePrediction,            BertForPreTraining,            BertForQuestionAnswering,            BertForSequenceClassification,            BertForTokenClassification,            BertLayer,            BertLMHeadModel,            BertModel,            BertPreTrainedModel,            load_tf_weights_in_bert,        )

然后点进去就可以看了,可以看他们的forward函数

免责声明:本网信息来自于互联网,目的在于传递更多信息,并不代表本网赞同其观点。其原创性以及文中陈述文字和内容未经本站证实,对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺,并请自行核实相关内容。本站不承担此类作品侵权行为的直接责任及连带责任。如若本网有任何内容侵犯您的权益,请及时联系我们,本站将会在24小时内处理完毕。
相关文章
返回顶部