Update BBSNet_model.py

BBSNet_model.py

@@ -297,8 +297,8 @@ class BBSNet(nn.Module):
         self.out1_conv = nn.Conv2d(32 * 2, 1, kernel_size=1, stride=1, bias=True)
         self.out2_conv = nn.Conv2d(32 * 1, 1, kernel_size=1, stride=1, bias=True)
 
-        if self.training:
-            self.initialize_weights()
+        #if self.training:
+        #    self.initialize_weights()
 
     def forward(self, x, x_depth):
         x = self.resnet.conv1(x)