kovacsvi commited on
Commit
73ee479
·
1 Parent(s): 75bb7a1

debug print for minor probs

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor_media.py +3 -1
interfaces/cap_minor_media.py CHANGED
@@ -118,7 +118,9 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
118
  with torch.no_grad():
119
  minor_logits = minor_model(**inputs).logits
120
  minor_probs = F.softmax(minor_logits, dim=-1)
121
-
 
 
122
  # Restrict to valid minor codes
123
  valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
124
  filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}
 
118
  with torch.no_grad():
119
  minor_logits = minor_model(**inputs).logits
120
  minor_probs = F.softmax(minor_logits, dim=-1)
121
+
122
+
123
+ print(minor_probs) # debug
124
  # Restrict to valid minor codes
125
  valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
126
  filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}