Merge pull request #129 from snakers4/adamnsandle

Adamnsandle
This commit is contained in:
Alexander Veysov
2021-12-07 15:40:07 +03:00
committed by GitHub
2 changed files with 41 additions and 9930 deletions

File diff suppressed because one or more lines are too long

View File

@@ -8,13 +8,10 @@ languages = ['ru', 'en', 'de', 'es']
def validate(model,
inputs: torch.Tensor,
**kwargs):
inputs: torch.Tensor):
with torch.no_grad():
outs = model(inputs, **kwargs)
if len(outs.shape) == 1:
return outs[1:]
return outs[:, 1] # 0 for noise, 1 for speech
outs = model(inputs)
return outs
def read_audio(path: str,