Commit 734d7acb authored by Andrei Roibu's avatar Andrei Roibu
Browse files

changed mse import

parent db5226f5
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment