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

Doc: loading from config alone does not load the model weights

Browse Source
This commit is contained in:
Lysandre
2019-08-19 10:17:47 -04:00
parent 5a49b793d9
commit c589862b78
8 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch_transformers/modeling_bert.py
View File

@@ -577,7 +577,9 @@ BERT_START_DOCSTRING = r""" The BERT model was proposed in
https://pytorch.org/docs/stable/nn.html#module
Parameters:
config (:class:`~pytorch_transformers.BertConfig`): Model configuration class with all the parameters of the model.
config (:class:`~pytorch_transformers.BertConfig`): Model configuration class with all the parameters of the model.
Initializing with a config file does not load the weights associated with the model, only the configuration.
Check out the :meth:`~pytorch_transformers.PreTrainedModel.from_pretrained` method to load the model weights.
"""
BERT_INPUTS_DOCSTRING = r"""