Segformer TF: fix output size in documentation (huggingface#18572)

* Segformer TF: fix output size in doc

* Segformer pytorch: fix output size in doc

Co-authored-by: Maxime Gardoni <[email protected]>

Author: 2 people

src/transformers/models/segformer/modeling_segformer.py

@@ -784,7 +784,7 @@ def forward(
 
         >>> inputs = feature_extractor(images=image, return_tensors="pt")
         >>> outputs = model(**inputs)
-        >>> logits = outputs.logits  # shape (batch_size, num_labels, height, width)
+        >>> logits = outputs.logits  # shape (batch_size, num_labels, height/4, width/4)
         >>> list(logits.shape)
         [1, 150, 128, 128]
         ```"""

src/transformers/models/segformer/modeling_tf_segformer.py

@@ -847,7 +847,7 @@ def call(
 
         >>> inputs = feature_extractor(images=image, return_tensors="tf")
         >>> outputs = model(**inputs, training=False)
-        >>> # logits are of shape (batch_size, num_labels, height, width)
+        >>> # logits are of shape (batch_size, num_labels, height/4, width/4)
         >>> logits = outputs.logits
         >>> list(logits.shape)
         [1, 150, 128, 128]