From 71030bb707494ee66d360bc14cb1e866e4991b4f Mon Sep 17 00:00:00 2001 From: Till Raab Date: Fri, 27 Oct 2023 11:22:00 +0200 Subject: [PATCH] embed for bf --- inference.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/inference.py b/inference.py index 22ed687..302d6f8 100644 --- a/inference.py +++ b/inference.py @@ -69,6 +69,8 @@ def main(args): inference_data = InferenceDataset(args.folder) inference_loader = create_inference_loader(inference_data) + embed() + quit() dataset_name = Path(args.folder).name infere_model(inference_loader, model, dataset_name)