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

rewrote 3D network, fixed memory bleed issues

parent 7d97cff9
......@@ -432,7 +432,7 @@ class BrainMapperUNet3D_Simple(nn.Module):
probability_map (torch.tensor): Output forward passed tensor through the U-net block
"""
Y_encoder_1, Y_np1, pool_indices1 = self.encoderBlock1.forward(X)
Y_encoder_1, Y_np1, pool_indices1 = self.encoderBlock1.forward(X)
Y_encoder_2, Y_np2, pool_indices2 = self.encoderBlock2.forward(
Y_encoder_1)
......
Supports Markdown
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