From fc3551d99ff6b962b0af019e1ff8173aa8478649 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Tue, 25 Jul 2023 17:38:04 +0200
Subject: [PATCH] Remove hardcoded predictions subfolder
---
src/ptbench/engine/callbacks.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/src/ptbench/engine/callbacks.py b/src/ptbench/engine/callbacks.py
index c2ce035f..2da897a7 100644
--- a/src/ptbench/engine/callbacks.py
+++ b/src/ptbench/engine/callbacks.py
@@ -407,8 +407,7 @@ class PredictionsWriter(lightning.pytorch.callbacks.BasePredictionWriter):
for dataloader_idx, dataloader_name in enumerate(dataloader_names):
logfile = os.path.join(
self.output_dir,
- f"predictions_{dataloader_name}",
- "predictions.csv",
+ f"{dataloader_name}.csv",
)
os.makedirs(os.path.dirname(logfile), exist_ok=True)
--
GitLab