diff --git a/src/transformers/models/segformer/modeling_segformer.py b/src/transformers/models/segformer/modeling_segformer.py
index b8be4cdb70a6d0..4e70eb957acf3f 100755
--- a/src/transformers/models/segformer/modeling_segformer.py
+++ b/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]
         ```"""
diff --git a/src/transformers/models/segformer/modeling_tf_segformer.py b/src/transformers/models/segformer/modeling_tf_segformer.py
index c2f4b2ff0c7cd8..2ff256d78d2b1d 100644
--- a/src/transformers/models/segformer/modeling_tf_segformer.py
+++ b/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]