Skip to content
This repository has been archived by the owner on Sep 11, 2022. It is now read-only.

Commit

Permalink
add copy conf for transformer_tts
Browse files Browse the repository at this point in the history
  • Loading branch information
yt605155624 committed Sep 27, 2021
1 parent 64fa640 commit be0c509
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions examples/transformer_tts/ljspeech/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
import argparse
import os
import logging
import shutil
from pathlib import Path

import jsonlines
Expand Down Expand Up @@ -198,8 +199,11 @@ 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)))
Expand Down

0 comments on commit be0c509

Please sign in to comment.