raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

embed for bf

Browse Source
This commit is contained in:
Till Raab 2023-10-27 11:22:00 +02:00
parent e3c0286d21
commit 71030bb707
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
View File

@ -69,6 +69,8 @@ def main(args):
inference_data = InferenceDataset(args.folder) inference_data = InferenceDataset(args.folder)
inference_loader = create_inference_loader(inference_data) inference_loader = create_inference_loader(inference_data)
embed()
quit()
dataset_name = Path(args.folder).name dataset_name = Path(args.folder).name
infere_model(inference_loader, model, dataset_name) infere_model(inference_loader, model, dataset_name)