changed mse import

solver.py

@@ -19,7 +19,6 @@ import glob
 from fsl.data.image import Image
 from fsl.utils.image.roi import roi
 from datetime import datetime
-from utils.losses import MSELoss
 from utils.common_utils import create_folder
 from utils.data_logging_utils import LogWriter
 from utils.early_stopping import EarlyStopping
@@ -64,7 +63,7 @@ class Solver():
                  experiment_name,
                  optimizer,
                  optimizer_arguments={},
-                 loss_function=MSELoss(),
+                 loss_function=torch.nn.MSELoss(),
                  model_name='BrainMapper',
                  labels=None,
                  number_epochs=10,