From 46aa1f5e7bdd6dfe5634fb7cf91fffcdc65f07b4 Mon Sep 17 00:00:00 2001
From: Sergey Polyakov <58026014+IamSVP94@users.noreply.github.com>
Date: Tue, 16 Mar 2021 16:35:29 +0300
Subject: [PATCH] Added dimension to tensor to save audio
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Иначе во всех примерах (в колабе по крайней мере), вылетала ошибка: ValueError: Expected 2D Tensor, got 1D.
---
 utils_vad.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils_vad.py b/utils_vad.py
index 0f2b46d..296cbf3 100644
--- a/utils_vad.py
+++ b/utils_vad.py
@@ -41,7 +41,7 @@ def read_audio(path: str,
 def save_audio(path: str,
                tensor: torch.Tensor,
                sr: int = 16000):
-    torchaudio.save(path, tensor, sr)
+    torchaudio.save(path, tensor.unsqueeze(0), sr)
 
 
 def init_jit_model(model_path: str,