"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ef8d6f2b4a2c3a5129438b7fa55ec29d4a4601ef"
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): ...@@ -849,8 +849,8 @@ class TFSegformerForSemanticSegmentation(TFSegformerPreTrainedModel):
>>> outputs = model(**inputs, training=False) >>> outputs = model(**inputs, training=False)
>>> # logits are of shape (batch_size, num_labels, height, width) >>> # logits are of shape (batch_size, num_labels, height, width)
>>> logits = outputs.logits >>> logits = outputs.logits
>>> logits.shape >>> list(logits.shape)
(1, 150, 128, 128) [1, 150, 128, 128]
```""" ```"""
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