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
51227e26
Unverified
Commit
51227e26
authored
Jul 29, 2022
by
Yih-Dar
Committed by
GitHub
Jul 29, 2022
Browse files
Fix TFSegformerForSemanticSegmentation doctest (#18362)
Co-authored-by:
ydshieh
<
ydshieh@users.noreply.github.com
>
parent
4e2f4a92
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
src/transformers/models/segformer/modeling_tf_segformer.py
src/transformers/models/segformer/modeling_tf_segformer.py
+2
-2
No files found.
src/transformers/models/segformer/modeling_tf_segformer.py
View file @
51227e26
...
...
@@ -849,8 +849,8 @@ class TFSegformerForSemanticSegmentation(TFSegformerPreTrainedModel):
>>> outputs = model(**inputs, training=False)
>>> # logits are of shape (batch_size, num_labels, height, width)
>>> logits = outputs.logits
>>> logits.shape
(
1, 150, 128, 128
)
>>>
list(
logits.shape
)
[
1, 150, 128, 128
]
```"""
return_dict
=
return_dict
if
return_dict
is
not
None
else
self
.
config
.
use_return_dict
output_hidden_states
=
(
...
...
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