commented out new weight initialisation for testing

BrainMapperAE.py

@@ -223,10 +223,10 @@ class BrainMapperAE3D(nn.Module):
                 for _, subsubmodule in submodule.named_children():
                     if isinstance(subsubmodule, (torch.nn.PReLU, torch.nn.Dropout3d, torch.nn.MaxPool3d)) == False:
                         subsubmodule.reset_parameters()
-                        if isinstance(subsubmodule, (torch.nn.Conv3d, torch.nn.ConvTranspose3d)):
-                            gain = np.sqrt(np.divide(2, 1 + np.power(0.25, 2)))
-                            fan, _ = calculate_fan(subsubmodule.weight)
-                            std = np.divide(gain, np.sqrt(fan))
-                            subsubmodule.weight.data.normal_(0, std)
+                        # if isinstance(subsubmodule, (torch.nn.Conv3d, torch.nn.ConvTranspose3d)):
+                        #     gain = np.sqrt(np.divide(2, 1 + np.power(0.25, 2)))
+                        #     fan, _ = calculate_fan(subsubmodule.weight)
+                        #     std = np.divide(gain, np.sqrt(fan))
+                        #     subsubmodule.weight.data.normal_(0, std)
 
-        print("Initialized network parameters!")
\ No newline at end of file
+        print("Initialized network parameters!")