go back to initial init

ahmedbesbes · Oct 21, 2019 · c4706be · c4706be
1 parent c161c67
commit c4706be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/model.py b/src/model.py
@@ -64,15 +64,15 @@ def __init__(self, args, number_of_classes):
 
         # initialize weights
 
-        # self._create_weights()
+        self._create_weights()
 
     # utility private functions
 
     def _create_weights(self, mean=0.0, std=0.05):
         for module in self.modules():
             if isinstance(module, nn.Conv1d) or isinstance(module, nn.Linear):
                 module.weight.data.normal_(mean, std)
-                
+
 
     def _get_conv_output(self, shape):
         x = torch.rand(shape)