AndreasXi commited on
Commit
9dace28
·
verified ·
1 Parent(s): 6cf10ea

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_finelap.py +6 -5
modeling_finelap.py CHANGED
@@ -121,11 +121,12 @@ class FineLAPModel(PreTrainedModel):
121
  global_text = self.get_global_text_embeds(text_labels, device)
122
 
123
  logits = torch.matmul(global_text, global_audio.transpose(-1, -2))
124
- if hasattr(self, "temp_global"):
125
- logits = logits / self.temp_global
126
- if hasattr(self, "b_global"):
127
- logits = logits + self.b_global
128
- return torch.sigmoid(logits).squeeze(-1)
 
129
 
130
  @torch.no_grad()
131
  def plot_frame_level_score(self, audio_path, text_labels, output_path="similarity_plot.png", device=None):
 
121
  global_text = self.get_global_text_embeds(text_labels, device)
122
 
123
  logits = torch.matmul(global_text, global_audio.transpose(-1, -2))
124
+ return logits
125
+ # if hasattr(self, "temp_global"):
126
+ # logits = logits / self.temp_global
127
+ # if hasattr(self, "b_global"):
128
+ # logits = logits + self.b_global
129
+ # return torch.sigmoid(logits).squeeze(-1)
130
 
131
  @torch.no_grad()
132
  def plot_frame_level_score(self, audio_path, text_labels, output_path="similarity_plot.png", device=None):