Pass task="binary" to binary metrics to handle latest torchmetrics (#920)

Author: oliverholworthy

merlin/models/torch/model/prediction_task.py

@@ -43,9 +43,9 @@ def build(self, input_size) -> SequentialBlock:
 class BinaryClassificationTask(PredictionTask):
     DEFAULT_LOSS = torch.nn.BCELoss()
     DEFAULT_METRICS = (
-        tm.Precision(num_classes=2),
-        tm.Recall(num_classes=2),
-        tm.Accuracy(),
+        tm.Precision(num_classes=2, task="binary"),
+        tm.Recall(num_classes=2, task="binary"),
+        tm.Accuracy(task="binary"),
         # TODO: Fix this: tm.AUC()
     )
 

requirements/pytorch.txt

@@ -1,2 +1,2 @@
 torch>=1.0
-torchmetrics==0.3.2
+torchmetrics>=0.10.0