File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
segment_anything/modeling Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -144,7 +144,7 @@ def __init__(
144
144
rel_pos_zero_init (bool): If True, zero initialize relative positional parameters.
145
145
window_size (int): Window size for window attention blocks. If it equals 0, then
146
146
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
148
148
parameter size.
149
149
"""
150
150
super ().__init__ ()
@@ -201,7 +201,7 @@ def __init__(
201
201
qkv_bias (bool: If True, add a learnable bias to query, key, value.
202
202
rel_pos (bool): If True, add relative positional embeddings to the attention map.
203
203
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
205
205
parameter size.
206
206
"""
207
207
super ().__init__ ()
You can’t perform that action at this time.
0 commit comments