Skip to content

Commit ae7c2fe

Browse files
author
tangy5
committed
change file mode
Signed-off-by: tangy5 <yucheng.tang@vanderbilt.edu>
1 parent d737121 commit ae7c2fe

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

monai/networks/blocks/text_embedding.py

100755100644
Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,12 +50,12 @@ def __init__(
5050

5151
def forward(self):
5252
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)
53+
test_encoding = nn.functional.relu(self.text_to_vision(self.text_embedding))
54+
test_encoding = test_encoding.unsqueeze(2).unsqueeze(2).unsqueeze(2)
5555
else:
5656
# text embedding as random initialized 'rand_embedding'
57-
task_encoding = self.text_embedding.weight.unsqueeze(2).unsqueeze(2).unsqueeze(2)
58-
return task_encoding
57+
test_encoding = self.text_embedding.weight.unsqueeze(2).unsqueeze(2).unsqueeze(2)
58+
return test_encoding
5959

6060

6161

0 commit comments

Comments
 (0)