"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "93f31e0e78e6d4bc7341ff3d34d60d78dafe1128"
Unverified Commit 6cb7d6ec authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

Fix doctest (#20534)


Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
parent d752337b
...@@ -474,11 +474,16 @@ class ResNetBackbone(ResNetPreTrainedModel): ...@@ -474,11 +474,16 @@ class ResNetBackbone(ResNetPreTrainedModel):
>>> image = Image.open(requests.get(url, stream=True).raw) >>> image = Image.open(requests.get(url, stream=True).raw)
>>> processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50") >>> processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
>>> model = AutoBackbone.from_pretrained("microsoft/resnet-50") >>> model = AutoBackbone.from_pretrained(
... "microsoft/resnet-50", out_features=["stage1", "stage2", "stage3", "stage4"]
... )
>>> inputs = processor(image, return_tensors="pt") >>> inputs = processor(image, return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)
>>> feature_maps = outputs.feature_maps
>>> list(feature_maps[-1].shape)
[1, 2048, 7, 7]
```""" ```"""
return_dict = return_dict if return_dict is not None else self.config.use_return_dict return_dict = return_dict if return_dict is not None else self.config.use_return_dict
output_hidden_states = ( output_hidden_states = (
......
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