Skip to content

Commit 33dc833

Browse files
Initialize label_attention_matrix to None before text_embedder branch
Co-authored-by: meilame-tayebjee <114609737+meilame-tayebjee@users.noreply.github.com>
1 parent f054a7f commit 33dc833

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

torchTextClassifiers/model/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,7 @@ def forward(
134134
Raw, not softmaxed.
135135
"""
136136
encoded_text = input_ids # clearer name
137+
label_attention_matrix = None
137138
if self.text_embedder is None:
138139
x_text = encoded_text.float()
139140
else:

0 commit comments

Comments
 (0)