Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
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