Spaces:

theAIguy
/

triplet_margin_loss

Sleeping

theAIguy commited on Jun 3, 2022

Commit

c224d5e

1 Parent(s): 2ea26d2

remove sequence

Files changed (1) hide show

triplet_margin_loss.py CHANGED Viewed

@@ -88,9 +88,9 @@ class TripletMarginLoss(evaluate.EvaluationModule):
             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
-                    "anchor": datasets.Sequence(datasets.Value("float")),
-                    "positive": datasets.Sequence(datasets.Value("float")),
-                    "negative": datasets.Sequence(datasets.Value("float")),
                     "margin": datasets.Value("float")
                 }
             ),

             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
+                    "anchor": datasets.Value("float"),
+                    "positive": datasets.Value("float"),
+                    "negative": datasets.Value("float"),
                     "margin": datasets.Value("float")
                 }
             ),