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

deleted enable_test_dropout - redundant function

parent 397a0fc0
......@@ -102,12 +102,6 @@ class BrainMapperUNet(nn.Module):
return probability_map
def enable_test_dropout(self):
"""
Description
"""
return None
def save(self, path):
"""
Description
......@@ -119,3 +113,21 @@ class BrainMapperUNet(nn.Module):
Description
"""
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