parent
aac76a1fb0
commit
b7f0bc74a7
|
@ -26,7 +26,7 @@ class MaskDecoder(nn.Module):
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Predicts masks given an image and prompt embeddings, using a
|
Predicts masks given an image and prompt embeddings, using a
|
||||||
tranformer architecture.
|
transformer architecture.
|
||||||
|
|
||||||
Arguments:
|
Arguments:
|
||||||
transformer_dim (int): the channel dimension of the transformer
|
transformer_dim (int): the channel dimension of the transformer
|
||||||
|
|
Loading…
Reference in New Issue