Adding NSFW Filter and the Style TTS2 Train Branch

2026-10-20 08:04:35 +00:00 · 2024-03-07 15:40:08 -05:00
parent 7053eec79c
commit a4bacef6e7
67 changed files with 178575 additions and 0 deletions
@@ -0,0 +1,14 @@
+def init_weights(m, mean=0.0, std=0.01):
+    classname = m.__class__.__name__
+    if classname.find("Conv") != -1:
+        m.weight.data.normal_(mean, std)
+
+
+def apply_weight_norm(m):
+    classname = m.__class__.__name__
+    if classname.find("Conv") != -1:
+        weight_norm(m)
+
+
+def get_padding(kernel_size, dilation=1):
+    return int((kernel_size*dilation - dilation)/2)