Added OG Repo need to make some more modifications

2026-10-28 09:08:54 +00:00 · 2024-03-08 23:40:11 -05:00
parent 772832eed6
commit 7137a25cf7
303 changed files with 168961 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)