Unverified Commit 51227e26 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix TFSegformerForSemanticSegmentation doctest (#18362)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 4e2f4a92
......@@ -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 = (
......
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