diff --git a/inference.py b/inference.py index 856e375..dceab85 100644 --- a/inference.py +++ b/inference.py @@ -69,7 +69,8 @@ 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)