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

Merge pull request #75 from Xrenya/patch-1

Browse Source 
Update docstring of image_encoder.py
This commit is contained in:
ericmintun 2023-04-10 08:54:57 -07:00 committed by GitHub
commit ca981bf377
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__()