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

bf

Browse Source
This commit is contained in:
Till Raab 2023-10-27 11:16:51 +02:00
parent 3690907f93
commit cbbf839e72
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
datasets.py
View File

@ -115,7 +115,7 @@ def create_valid_loader(valid_dataset, num_workers=0):
) )
return valid_loader return valid_loader
def create_inference_loader(inference_dataset, num_workers=0) def create_inference_loader(inference_dataset, num_workers=0):
inference_loader = DataLoader( inference_loader = DataLoader(
inference_dataset, inference_dataset,
batch_size=BATCH_SIZE, batch_size=BATCH_SIZE,