Merge pull request #186 from yt605155624/fix_transformertts

fix conf copy

examples/transformer_tts/ljspeech/train.py

@@ -153,6 +153,10 @@ def train_sp(args, config):
     output_dir = Path(args.output_dir)
     output_dir.mkdir(parents=True, exist_ok=True)
 
+    config_name = args.config.split("/")[-1]
+    # copy conf to output_dir
+    shutil.copyfile(args.config, output_dir / config_name)
+
     updater = TransformerTTSUpdater(
         model=model,
         optimizer=optimizer,
@@ -199,11 +203,9 @@ def main():
     if args.device == "cpu" and args.nprocs > 1:
         raise RuntimeError("Multiprocess training on CPU is not supported.")
     config = get_cfg_default()
-    config_name = args.config.split("/")[-1]
+
     if args.config:
         config.merge_from_file(args.config)
-    # copy conf to output_dir
-    shutil.copyfile(args.config, Path(args.output_dir) / config_name)
 
     print("========Args========")
     print(yaml.safe_dump(vars(args)))