zjt/SAMapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update docstring of image_encoder.py

Browse Source 
minor docstring fix
This commit is contained in:
Rinat 2023-04-07 16:49:32 +09:00 committed by GitHub
parent 9e1eb9fdbc
commit a77cb1fda9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
segment_anything/modeling/image_encoder.py
View File

@ -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__()