Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Andrei-Claudiu Roibu
BrainMapper
Commits
42ae0d12
Commit
42ae0d12
authored
Mar 25, 2020
by
Andrei-Claudiu Roibu
🖥
Browse files
deleted enable_test_dropout - redundant function
parent
397a0fc0
Changes
1
Show whitespace changes
Inline
Side-by-side
BrainMapperUNet.py
View file @
42ae0d12
...
...
@@ -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
()
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment