AB739 commited on
Commit
f5f0a79
·
verified ·
1 Parent(s): c10b387

Update tasks/audio.py

Browse files
Files changed (1) hide show
  1. tasks/audio.py +3 -3
tasks/audio.py CHANGED
@@ -68,9 +68,9 @@ async def evaluate_audio(request: AudioEvaluationRequest):
68
  test_loader = DataLoader(
69
  TensorDataset(waveforms, labels),
70
  batch_size=128,
71
- shuffle=False,
72
- pin_memory=True,
73
- num_workers=4
74
  )
75
 
76
  # Load Openvino model
 
68
  test_loader = DataLoader(
69
  TensorDataset(waveforms, labels),
70
  batch_size=128,
71
+ shuffle=False
72
+ #pin_memory=True,
73
+ #num_workers=4
74
  )
75
 
76
  # Load Openvino model