Skip to content
Snippets Groups Projects
Commit 5cb7b7c5 authored by chinyun_yu_joey's avatar chinyun_yu_joey
Browse files

use batch size = 4

parent d1221dbd
No related branches found
Tags submission-demucs-split
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
test_demucs.py
+ 1
− 1
View file @ 5cb7b7c5
...@@ -19,7 +19,7 @@ def separate( ...@@ -19,7 +19,7 @@ def separate(
# convert numpy audio to torch # convert numpy audio to torch
audio_torch = torch.tensor(audio.T).float().to(device) audio_torch = torch.tensor(audio.T).float().to(device)
max_shift = int(rate * 0.5) max_shift = int(rate * 0.5)
batch_size = 3 batch_size = 4
valid_length = audio_torch.size(1) valid_length = audio_torch.size(1)
padded_length = valid_length + max_shift padded_length = valid_length + max_shift
padded_audio = F.pad(audio_torch, [max_shift] * 2) padded_audio = F.pad(audio_torch, [max_shift] * 2)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment