Skip to content

GitLab

Unverified Commit cf8a7c24 authored by Mishig Davaadorj's avatar Mishig Davaadorj Committed by GitHub
Browse files

Update custom_models.mdx (#16964) 

BertModelForSequenceClassification -> BertForSequenceClassification
parent 5896b3ec
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/en/custom_models.mdx
View file @ cf8a7c24
...@@ -106,7 +106,7 @@ directly upload your config to the Hub. ...@@ -106,7 +106,7 @@ directly upload your config to the Hub.
Now that we have our ResNet configuration, we can go on writing the model. We will actually write two: one that Now that we have our ResNet configuration, we can go on writing the model. We will actually write two: one that
extracts the hidden features from a batch of images (like [`BertModel`]) and one that is suitable for image extracts the hidden features from a batch of images (like [`BertModel`]) and one that is suitable for image
classification (like [`BertModelForSequenceClassification`]). classification (like [`BertForSequenceClassification`]).
As we mentioned before, we'll only write a loose wrapper of the model to keep it simple for this example. The only As we mentioned before, we'll only write a loose wrapper of the model to keep it simple for this example. The only
thing we need to do before writing this class is a map between the block types and actual block classes. Then the thing we need to do before writing this class is a map between the block types and actual block classes. Then the
......
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