Commit 42ae0d12 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

deleted enable_test_dropout - redundant function

parent 397a0fc0
...@@ -101,12 +101,6 @@ class BrainMapperUNet(nn.Module): ...@@ -101,12 +101,6 @@ class BrainMapperUNet(nn.Module):
probability_map = self.classifier.forward(Y_decoder_4) probability_map = self.classifier.forward(Y_decoder_4)
return probability_map return probability_map
def enable_test_dropout(self):
"""
Description
"""
return None
def save(self, path): def save(self, path):
""" """
...@@ -119,3 +113,21 @@ class BrainMapperUNet(nn.Module): ...@@ -119,3 +113,21 @@ class BrainMapperUNet(nn.Module):
Description Description
""" """
return None return None
if __name__ == '__main__':
parameters = {
'kernel_heigth': 5,
'kernel_width': 5,
'kernel_classification': 1,
'input_channels': 1,
'output_channels': 64,
'convolution_stride': 1,
'dropout': 0.2,
'pool_kernel_size': 2,
'pool_stride': 2,
'up_mode': 'upconv',
'number_of_classes': 1
}
network = BrainMapperUNet(parameters)
network.enable_test_dropout()
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