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
68fc4520
Commit
68fc4520
authored
Aug 07, 2020
by
Andrei Roibu
Browse files
added argument for using last checkpoing in command line
parent
41956de8
Changes
1
Hide whitespace changes
Inline
Side-by-side
run.py
View file @
68fc4520
...
...
@@ -282,6 +282,8 @@ if __name__ == '__main__':
help
=
'run mode, valid values are train or evaluate'
)
parser
.
add_argument
(
'--model_name'
,
'-n'
,
required
=
True
,
help
=
'model name, required for identifying the settings file modelName.ini & modelName_eval.ini'
)
parser
.
add_argument
(
'--use_last_checkpoint'
,
'-cp'
,
required
=
False
,
help
=
'flag indicating if the last checkpoint should be used if 1; useful when wanting to time-limit jobs.'
)
arguments
=
parser
.
parse_args
()
...
...
@@ -295,6 +297,9 @@ if __name__ == '__main__':
misc_parameters
=
settings
[
'MISC'
]
evaluation_parameters
=
settings
[
'EVALUATION'
]
if
arguments
.
use_last_checkpoint
is
not
None
:
training_parameters
[
'use_last_checkpoint'
]
=
True
if
arguments
.
mode
==
'train'
:
train
(
data_parameters
,
training_parameters
,
network_parameters
,
misc_parameters
)
...
...
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