remove init, because empiricism

Author: lucidrains

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)

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',