From 1f52eb0eb71c2c32a9a7fb871479cc98be9198fd Mon Sep 17 00:00:00 2001
From: PluieElectrique <41453973+PluieElectrique@users.noreply.github.com>
Date: Wed, 22 Jun 2022 03:27:43 +0000
Subject: [PATCH] Fix `get_num_classes` error

---
 requirements/requirements.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index ac7d1a3dd472..3f34a97142a8 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -1,5 +1,6 @@
 numpy>=1.18.2
 onnx>=1.7.0
+torchmetrics<=0.7.3
 pytorch-lightning>=1.3.0
 python-dateutil
 torch