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 d737121 commit ae7c2feCopy full SHA for ae7c2fe
monai/networks/blocks/text_embedding.py
100755
100644
@@ -50,12 +50,12 @@ def __init__(
50
51
def forward(self):
52
if self.encoding == 'clip_embedding':
53
- task_encoding = nn.functional.relu(self.text_to_vision(self.text_embedding))
54
- task_encoding = task_encoding.unsqueeze(2).unsqueeze(2).unsqueeze(2)
+ test_encoding = nn.functional.relu(self.text_to_vision(self.text_embedding))
+ test_encoding = test_encoding.unsqueeze(2).unsqueeze(2).unsqueeze(2)
55
else:
56
# text embedding as random initialized 'rand_embedding'
57
- task_encoding = self.text_embedding.weight.unsqueeze(2).unsqueeze(2).unsqueeze(2)
58
- return task_encoding
+ test_encoding = self.text_embedding.weight.unsqueeze(2).unsqueeze(2).unsqueeze(2)
+ return test_encoding
59
60
61
0 commit comments