Unverified Commit 9ad77aff authored by Francesco Saverio Zuppichini's avatar Francesco Saverio Zuppichini Committed by GitHub
Browse files

test (#16294)

parent d50f62f2
...@@ -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",
......
...@@ -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 = [
......
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