Commit b1633c87 authored by Andrei Roibu's avatar Andrei Roibu
Browse files

deleted print functions

parent 9a088f2f
......@@ -188,9 +188,6 @@ class Solver():
# We add an extra dimension (~ number of channels) for the 3D convolutions.
X = torch.unsqueeze(X, dim=1)
y = torch.unsqueeze(y, dim=1)
print('X range:', torch.min(X), torch.max(X))
print('y range:', torch.min(y), torch.max(y))
MNI152_T1_2mm_brain_mask = torch.unsqueeze(torch.unsqueeze(self.MNI152_T1_2mm_brain_mask, dim=0), dim=0)
......@@ -202,12 +199,8 @@ class Solver():
y_hat = model(X) # Forward pass & Masking
print('y_hat range:', torch.min(y_hat), torch.max(y_hat))
y_hat = torch.mul(y_hat, MNI152_T1_2mm_brain_mask)
print('y_hat masked range:', torch.min(y_hat), torch.max(y_hat))
loss = self.loss_function(y_hat, y) # Loss computation
if phase == 'train':
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