From 7a03f1bec7001d92e2fe8a71c7437d6848eb88b2 Mon Sep 17 00:00:00 2001 From: YuriHead Date: Fri, 14 Apr 2023 10:54:34 +0800 Subject: [PATCH] Update utils.py --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils.py b/utils.py index fb60150..960b8f8 100644 --- a/utils.py +++ b/utils.py @@ -232,7 +232,7 @@ def get_hubert_content(hmodel, wav_16k_tensor): inputs = { "source": feats.to(wav_16k_tensor.device), "padding_mask": padding_mask.to(wav_16k_tensor.device), - "output_layer": 9, # layer 9 + "output_layer": 12, # layer 12 } with torch.no_grad(): logits = hmodel.extract_features(**inputs)