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
9ad77aff
Unverified
Commit
9ad77aff
authored
Mar 21, 2022
by
Francesco Saverio Zuppichini
Committed by
GitHub
Mar 21, 2022
Browse files
test (#16294)
parent
d50f62f2
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
src/transformers/models/resnet/modeling_resnet.py
src/transformers/models/resnet/modeling_resnet.py
+3
-3
src/transformers/models/van/modeling_van.py
src/transformers/models/van/modeling_van.py
+1
-1
No files found.
src/transformers/models/resnet/modeling_resnet.py
View file @
9ad77aff
...
...
@@ -37,12 +37,12 @@ _CONFIG_FOR_DOC = "ResNetConfig"
_FEAT_EXTRACTOR_FOR_DOC
=
"AutoFeatureExtractor"
# Base docstring
_CHECKPOINT_FOR_DOC
=
""
_CHECKPOINT_FOR_DOC
=
"
microsoft/resnet-50
"
_EXPECTED_OUTPUT_SHAPE
=
[
1
,
2048
,
7
,
7
]
# Image classification docstring
_IMAGE_CLASS_CHECKPOINT
=
""
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"
'tabby, tabby
cat
'
"
_IMAGE_CLASS_CHECKPOINT
=
"
microsoft/resnet-50
"
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"
tiger
cat"
RESNET_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
"microsoft/resnet-50"
,
...
...
src/transformers/models/van/modeling_van.py
View file @
9ad77aff
...
...
@@ -47,7 +47,7 @@ _CHECKPOINT_FOR_DOC = "van-base"
_EXPECTED_OUTPUT_SHAPE
=
[
1
,
512
,
7
,
7
]
# Image classification docstring
_IMAGE_CLASS_CHECKPOINT
=
"van-base"
_IMAGE_CLASS_CHECKPOINT
=
"
Visual-Attention-Network/
van-base"
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"tabby, tabby cat"
VAN_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
...
...
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