From e36681ef56efbdeeb654ca69bb3ab176cd857724 Mon Sep 17 00:00:00 2001 From: Utkarsh Yashwant Tambe <89902664+utkarshtambe10@users.noreply.github.com> Date: Thu, 27 Jun 2024 15:18:16 +0530 Subject: [PATCH] adding torchmetrics to requirements The library torchmetrics has not been mentioned in the requirements, so adding it. --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 2005b95..4e91950 100644 --- a/README.md +++ b/README.md @@ -55,6 +55,7 @@ Skin lesion segmentation plays a critical role in the early detection and accura scipy==1.6.3 termcolor==2.3.0 torch==2.0.1 + torchmetrics==1.4.0.post0 torchvision==0.15.2 tqdm==4.65.0 ```