Unverified Commit 7ecff0cc authored by ELanning's avatar ELanning Committed by GitHub
Browse files

Fix typo in training (#5510)

parent 58cca47c
...@@ -39,7 +39,7 @@ of the specified model are used to initialize the model. The ...@@ -39,7 +39,7 @@ of the specified model are used to initialize the model. The
library also includes a number of task-specific final layers or 'heads' whose library also includes a number of task-specific final layers or 'heads' whose
weights are instantiated randomly when not present in the specified weights are instantiated randomly when not present in the specified
pre-trained model. For example, instantiating a model with pre-trained model. For example, instantiating a model with
``BertForSequenceClassification.from_pretrained('bert-base-uncased', num_classes=2)`` ``BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=2)``
will create a BERT model instance with encoder weights copied from the will create a BERT model instance with encoder weights copied from the
``bert-base-uncased`` model and a randomly initialized sequence ``bert-base-uncased`` model and a randomly initialized sequence
classification head on top of the encoder with an output size of 2. Models classification head on top of the encoder with an output size of 2. Models
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment