From e3ccd3b135575a78f609f099a5d1e1169af3d098 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Thu, 22 Jun 2023 14:49:00 +0200
Subject: [PATCH] Updated pyproject.toml
---
pyproject.toml | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pyproject.toml b/pyproject.toml
index 6b381da2..e43899d2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -41,7 +41,7 @@ dependencies = [
"pillow",
"torch>=1.8",
"torchvision>=0.10",
- "lightning",
+ "lightning>=2.0.3",
"tensorboard",
]
@@ -117,8 +117,8 @@ montgomery_rs_f7 = "ptbench.configs.datasets.montgomery_RS.fold_7"
montgomery_rs_f8 = "ptbench.configs.datasets.montgomery_RS.fold_8"
montgomery_rs_f9 = "ptbench.configs.datasets.montgomery_RS.fold_9"
# shenzhen dataset (and cross-validation folds)
-shenzhen = "ptbench.configs.datasets.shenzhen.default"
-shenzhen_rgb = "ptbench.configs.datasets.shenzhen.rgb"
+shenzhen = "ptbench.data.shenzhen.default"
+shenzhen_rgb = "ptbench.data.shenzhen.rgb"
shenzhen_f0 = "ptbench.data.shenzhen.fold_0"
shenzhen_f1 = "ptbench.data.shenzhen.fold_1"
shenzhen_f2 = "ptbench.data.shenzhen.fold_2"
--
GitLab