From 7e69dd49242e7f136b8c00bc411938fd54c29cf8 Mon Sep 17 00:00:00 2001 From: peterdudfield Date: Fri, 9 Feb 2024 09:47:59 +0000 Subject: [PATCH] torchvision==0.16.2 --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 574b5a2..051c716 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ pydantic +torchvision==0.16.2 pytorch-lightning==2.1.3 torch[cpu]==2.2.0 PVNet-summation==0.1.2