Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
cf8a7c24
Unverified
Commit
cf8a7c24
authored
Apr 27, 2022
by
Mishig Davaadorj
Committed by
GitHub
Apr 27, 2022
Browse files
Update custom_models.mdx (#16964)
BertModelForSequenceClassification -> BertForSequenceClassification
parent
5896b3ec
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
docs/source/en/custom_models.mdx
docs/source/en/custom_models.mdx
+1
-1
No files found.
docs/source/en/custom_models.mdx
View file @
cf8a7c24
...
...
@@ -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
extracts
the
hidden
features
from
a
batch
of
images
(
like
[`
BertModel
`])
and
one
that
is
suitable
for
image
classification
(
like
[`
Bert
Model
ForSequenceClassification
`]).
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
thing we need to do before writing this class is a map between the block types and actual block classes. Then the
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment