logit

classpred.py

@@ -39,6 +39,6 @@ def predict_class(x, sr, start, end):
         logits = MODEL(x.view(1, 1, 1024, 128)).squeeze(0)
 
     topk_probs, topk_classes = logits.sigmoid().topk(10)
-    preds = [[AUDIOSET_LABELS[cls], prob.item() * 100] for cls, prob in zip(topk_classes, topk_probs)]
+    preds = [[AUDIOSET_LABELS[cls], prob.item()] for cls, prob in zip(topk_classes, topk_probs)]
 
     return preds