Skip to content

Commit a77cb1f

Browse files
authored
Update docstring of image_encoder.py
minor docstring fix
1 parent 9e1eb9f commit a77cb1f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

segment_anything/modeling/image_encoder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def __init__(
144144
rel_pos_zero_init (bool): If True, zero initialize relative positional parameters.
145145
window_size (int): Window size for window attention blocks. If it equals 0, then
146146
use global attention.
147-
input_size (int or None): Input resolution for calculating the relative positional
147+
input_size (tuple(int, int) or None): Input resolution for calculating the relative positional
148148
parameter size.
149149
"""
150150
super().__init__()
@@ -201,7 +201,7 @@ def __init__(
201201
qkv_bias (bool: If True, add a learnable bias to query, key, value.
202202
rel_pos (bool): If True, add relative positional embeddings to the attention map.
203203
rel_pos_zero_init (bool): If True, zero initialize relative positional parameters.
204-
input_size (int or None): Input resolution for calculating the relative positional
204+
input_size (tuple(int, int) or None): Input resolution for calculating the relative positional
205205
parameter size.
206206
"""
207207
super().__init__()

0 commit comments

Comments
 (0)