mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow starting step to be forced

Browse Source 
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
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/train.py
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))