embed for bf
This commit is contained in:
parent
e3c0286d21
commit
71030bb707
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user