diff --git a/codes/train.py b/codes/train.py index 980c6ffc..8d7551d9 100644 --- a/codes/train.py +++ b/codes/train.py @@ -300,7 +300,7 @@ class Trainer: if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_clip_cond_to_voice.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_diffusion_tts_medium.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0) args = parser.parse_args() diff --git a/codes/trainer/ExtensibleTrainer.py b/codes/trainer/ExtensibleTrainer.py index e12253e5..7e2d56f8 100644 --- a/codes/trainer/ExtensibleTrainer.py +++ b/codes/trainer/ExtensibleTrainer.py @@ -161,7 +161,7 @@ class ExtensibleTrainer(BaseModel): # Load experiments self.experiments = [] if 'experiments' in opt.keys(): - self.experiments = [get_experiment_for_name(e) for e in op['experiments']] + self.experiments = [get_experiment_for_name(e) for e in opt['experiments']] # Setting this to false triggers SRGAN to call the models update_model() function on the first iteration. self.updated = True