Allow starting step to be forced

Useful for testing purposes or to force a validation.
This commit is contained in:
James Betker 2020-10-19 15:23:04 -06:00
parent 9b9a6e5925
commit 331c40f0c8

View File

@ -170,6 +170,8 @@ def main():
else: else:
current_step = -1 if 'start_step' not in opt.keys() else opt['start_step'] current_step = -1 if 'start_step' not in opt.keys() else opt['start_step']
start_epoch = 0 start_epoch = 0
if 'force_start_step' in opt.keys():
current_step = opt['force_start_step']
#### training #### training
logger.info('Start training from epoch: {:d}, iter: {:d}'.format(start_epoch, current_step)) logger.info('Start training from epoch: {:d}, iter: {:d}'.format(start_epoch, current_step))