From 80929e472e2b0590c3ef8984690d2c950fffc797 Mon Sep 17 00:00:00 2001
From: ms903x1 <108673569+ms903x1@users.noreply.github.com>
Date: Fri, 2 Jun 2023 10:27:20 +0800
Subject: [PATCH] Update data_utils.py (#407)

* Update envfilescheck.bat

add pretrained_v2 and uvr5 update

* Update envfilescheck.bat

fix bug

* Update envfilescheck.bat

fix bug

* Update data_utils.py

fix bug where data exceeding 4s is filtered out.
---
 train/data_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train/data_utils.py b/train/data_utils.py
index 6e00a7a..af044d0 100644
--- a/train/data_utils.py
+++ b/train/data_utils.py
@@ -38,7 +38,7 @@ class TextAudioLoaderMultiNSFsid(torch.utils.data.Dataset):
         for audiopath, text, pitch, pitchf, dv in self.audiopaths_and_text:
             if self.min_text_len <= len(text) and len(text) <= self.max_text_len:
                 audiopaths_and_text_new.append([audiopath, text, pitch, pitchf, dv])
-                lengths.append(os.path.getsize(audiopath) // (2 * self.hop_length))
+                lengths.append(os.path.getsize(audiopath) // (3 * self.hop_length))
         self.audiopaths_and_text = audiopaths_and_text_new
         self.lengths = lengths
 
@@ -246,7 +246,7 @@ class TextAudioLoader(torch.utils.data.Dataset):
         for audiopath, text, dv in self.audiopaths_and_text:
             if self.min_text_len <= len(text) and len(text) <= self.max_text_len:
                 audiopaths_and_text_new.append([audiopath, text, dv])
-                lengths.append(os.path.getsize(audiopath) // (2 * self.hop_length))
+                lengths.append(os.path.getsize(audiopath) // (3 * self.hop_length))
         self.audiopaths_and_text = audiopaths_and_text_new
         self.lengths = lengths