diff --git a/pyproject.toml b/pyproject.toml
index edffb97..858d9d7 100755
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,6 +40,7 @@ dependencies = [
   "pillow",
   "torch",
   "torchvision",
+  "torchmetrics",
   "dotmap",
   "loguru",
   "jinja2",