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
507756d2
Commit
507756d2
authored
Aug 14, 2020
by
Andrei Roibu
Browse files
added adamw + minus one scaling flags
parent
609a4565
Changes
1
Hide whitespace changes
Inline
Side-by-side
run.py
View file @
507756d2
...
...
@@ -240,8 +240,10 @@ def train(data_parameters, training_parameters, network_parameters, misc_paramet
return
validation_loss
optimizer
=
torch
.
optim
.
Adam
# optimizer = torch.optim.AdamW
if
training_parameters
[
'adam_w_flag'
]
==
True
:
optimizer
=
torch
.
optim
.
AdamW
else
:
optimizer
=
torch
.
optim
.
Adam
loss_function
=
torch
.
nn
.
MSELoss
()
# loss_function=torch.nn.L1Loss()
...
...
@@ -334,6 +336,7 @@ def evaluate_mapping(mapping_evaluation_parameters):
mean_subtraction_flag
=
mapping_evaluation_parameters
[
'mean_subtraction_flag'
]
scale_volumes_flag
=
mapping_evaluation_parameters
[
'scale_volumes_flag'
]
normalize_flag
=
mapping_evaluation_parameters
[
'normalize_flag'
]
minus_one_scaling_flag
=
mapping_evaluation_parameters
[
'minus_one_scaling_flag'
]
negative_flag
=
mapping_evaluation_parameters
[
'negative_flag'
]
outlier_flag
=
mapping_evaluation_parameters
[
'outlier_flag'
]
shrinkage_flag
=
mapping_evaluation_parameters
[
'shrinkage_flag'
]
...
...
@@ -354,6 +357,7 @@ def evaluate_mapping(mapping_evaluation_parameters):
mean_subtraction_flag
,
scale_volumes_flag
,
normalize_flag
,
minus_one_scaling_flag
,
negative_flag
,
outlier_flag
,
shrinkage_flag
,
...
...
Write
Preview
Markdown
is supported
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