a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

update training data encountered when using force_start_step

Browse Source
This commit is contained in:
James Betker 2022-04-04 12:25:00 -06:00
parent cdd12ff46c
commit 4cdb0169d0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/train.py
View File

@ -165,6 +165,7 @@ class Trainer:
self.start_epoch = 0 self.start_epoch = 0
if 'force_start_step' in opt.keys(): if 'force_start_step' in opt.keys():
self.current_step = opt['force_start_step'] self.current_step = opt['force_start_step']
self.total_training_data_encountered = self.current_step * opt['datasets']['train']['batch_size']
opt['current_step'] = self.current_step opt['current_step'] = self.current_step
def do_step(self, train_data): def do_step(self, train_data):
@ -318,7 +319,7 @@ class Trainer:
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../experiments/train_autoregressive_codegen.yml') parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_clip_text_to_voice.yml')
parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
args = parser.parse_args() args = parser.parse_args()
opt = option.parse(args.opt, is_train=True) opt = option.parse(args.opt, is_train=True)