added save function

BrainMapperUNet.py

@@ -103,10 +103,23 @@ class BrainMapperUNet(nn.Module):
         return probability_map
     
     def save(self, path):
+        """Model Saver
+        
+        Function saving the model with all its parameters to a given path.
+        The path must end with a *.model argument.
+
+        Args:
+            path (str): Path string
+
+        Returns:
+            None
+        
+        Raises:
+            None
         """
-        Description
-        """
-        pass
+        
+        print("Saving Model... {}".format(path))
+        torch.save(self, path)
         
     def predict(self, X):
         """