AB739 commited on
Commit
9822092
·
verified ·
1 Parent(s): bc54a04

Update tasks/audio.py

Browse files
Files changed (1) hide show
  1. tasks/audio.py +1 -1
tasks/audio.py CHANGED
@@ -82,7 +82,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
82
 
83
  test_loader = DataLoader(
84
  TensorDataset(waveforms, labels),
85
- batch_size=64,
86
  shuffle=False,
87
  pin_memory=True,
88
  num_workers=4
 
82
 
83
  test_loader = DataLoader(
84
  TensorDataset(waveforms, labels),
85
+ batch_size=128,
86
  shuffle=False,
87
  pin_memory=True,
88
  num_workers=4