remove init, because empiricism

lucidrains · Sep 18, 2021 · f8e2f2d · f8e2f2d
1 parent fe9b20f
commit f8e2f2d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/se3_transformer_pytorch/se3_transformer_pytorch.py b/se3_transformer_pytorch/se3_transformer_pytorch.py
@@ -295,12 +295,6 @@ def __init__(
             nn.Linear(mid_dim, num_freq * in_dim * out_dim)
         )
 
-        self.apply(self.init_)
-
-    def init_(self, m):
-        if isinstance(m, nn.Linear):
-            nn.init.kaiming_uniform_(m.weight)
-
     def forward(self, x):
         y = self.net(x)
         return rearrange(y, '... (o i f) -> ... o () i () f', i = self.in_dim, o = self.out_dim)

diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
   name = 'se3-transformer-pytorch',
   packages = find_packages(),
   include_package_data = True,
-  version = '0.8.12',
+  version = '0.8.13',
   license='MIT',
   description = 'SE3 Transformer - Pytorch',
   author = 'Phil Wang',