From 0befddab8dadfc387cbf09a86651a35f3de235d3 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 11:32:48 +0200
Subject: [PATCH] bf
---
inference.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/inference.py b/inference.py
index dceab85..a09f0d2 100644
--- a/inference.py
+++ b/inference.py
@@ -69,10 +69,9 @@ def main(args):
inference_loader = create_inference_loader(inference_data)
dataset_name = Path(args.folder).name
- embed()
- quit()
- if not (Path(OUTDIR)/dataset_name).exists():
- Path(Path(OUTDIR)/dataset_name).mkdir(parents=True, exist_ok=True)
+
+ if not (Path(INFERENCE_OUTDIR)/dataset_name).exists():
+ Path(Path(INFERENCE_OUTDIR)/dataset_name).mkdir(parents=True, exist_ok=True)
infere_model(inference_loader, model, dataset_name)