We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae7c2fe commit ec603d7Copy full SHA for ec603d7
monai/networks/blocks/text_embedding.py
@@ -24,7 +24,7 @@ class TextEncoder(nn.Module):
24
25
Connecting text and medical 3D image, based on: "Liu et al.,
26
CLIP-Driven Universal Model for Organ Segmentation and Tumor Detection <https://arxiv.org/pdf/2301.00785.pdf>"
27
- """
+ """
28
def __init__(
29
self,
30
out_channels: int,
@@ -56,6 +56,3 @@ def forward(self):
56
# text embedding as random initialized 'rand_embedding'
57
test_encoding = self.text_embedding.weight.unsqueeze(2).unsqueeze(2).unsqueeze(2)
58
return test_encoding
59
-
60
61
0 commit comments