Commit ff37a962 authored by Andrei Roibu's avatar Andrei Roibu
Browse files

added command line argument for using last checkpoint

parent 40295fb1
......@@ -389,6 +389,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()
......@@ -402,6 +404,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)
......
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