From a77cb1fda95f4288343ba7cbd2cb802ce359772b Mon Sep 17 00:00:00 2001 From: Rinat <51479797+Xrenya@users.noreply.github.com> Date: Fri, 7 Apr 2023 16:49:32 +0900 Subject: [PATCH] Update docstring of image_encoder.py minor docstring fix --- segment_anything/modeling/image_encoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/segment_anything/modeling/image_encoder.py b/segment_anything/modeling/image_encoder.py index a6ad9ad..736870f 100644 --- a/segment_anything/modeling/image_encoder.py +++ b/segment_anything/modeling/image_encoder.py @@ -144,7 +144,7 @@ class Block(nn.Module): rel_pos_zero_init (bool): If True, zero initialize relative positional parameters. window_size (int): Window size for window attention blocks. If it equals 0, then use global attention. - input_size (int or None): Input resolution for calculating the relative positional + input_size (tuple(int, int) or None): Input resolution for calculating the relative positional parameter size. """ super().__init__() @@ -201,7 +201,7 @@ class Attention(nn.Module): qkv_bias (bool: If True, add a learnable bias to query, key, value. rel_pos (bool): If True, add relative positional embeddings to the attention map. rel_pos_zero_init (bool): If True, zero initialize relative positional parameters. - input_size (int or None): Input resolution for calculating the relative positional + input_size (tuple(int, int) or None): Input resolution for calculating the relative positional parameter size. """ super().__init__()