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
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
Hide 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"
...
@@ -37,12 +37,12 @@ _CONFIG_FOR_DOC = "ResNetConfig"
_FEAT_EXTRACTOR_FOR_DOC
=
"AutoFeatureExtractor"
_FEAT_EXTRACTOR_FOR_DOC
=
"AutoFeatureExtractor"
# Base docstring
# Base docstring
_CHECKPOINT_FOR_DOC
=
""
_CHECKPOINT_FOR_DOC
=
"
microsoft/resnet-50
"
_EXPECTED_OUTPUT_SHAPE
=
[
1
,
2048
,
7
,
7
]
_EXPECTED_OUTPUT_SHAPE
=
[
1
,
2048
,
7
,
7
]
# Image classification docstring
# Image classification docstring
_IMAGE_CLASS_CHECKPOINT
=
""
_IMAGE_CLASS_CHECKPOINT
=
"
microsoft/resnet-50
"
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"
'tabby, tabby
cat
'
"
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"
tiger
cat"
RESNET_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
RESNET_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
"microsoft/resnet-50"
,
"microsoft/resnet-50"
,
...
...
src/transformers/models/van/modeling_van.py
View file @
9ad77aff
...
@@ -47,7 +47,7 @@ _CHECKPOINT_FOR_DOC = "van-base"
...
@@ -47,7 +47,7 @@ _CHECKPOINT_FOR_DOC = "van-base"
_EXPECTED_OUTPUT_SHAPE
=
[
1
,
512
,
7
,
7
]
_EXPECTED_OUTPUT_SHAPE
=
[
1
,
512
,
7
,
7
]
# Image classification docstring
# Image classification docstring
_IMAGE_CLASS_CHECKPOINT
=
"van-base"
_IMAGE_CLASS_CHECKPOINT
=
"
Visual-Attention-Network/
van-base"
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"tabby, tabby cat"
_IMAGE_CLASS_EXPECTED_OUTPUT
=
"tabby, tabby cat"
VAN_PRETRAINED_MODEL_ARCHIVE_LIST
=
[
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