embed for bf

This commit is contained in:
Till Raab 2023-10-27 11:22:00 +02:00
parent e3c0286d21
commit 71030bb707

View File

@ -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)