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

Commit

Permalink
Merge pull request #184 from yt605155624/transformer_tts_finetune
Browse files Browse the repository at this point in the history
add copy conf for transformer_tts
  • Loading branch information
yt605155624 authored Sep 27, 2021
2 parents 062b0bd + be0c509 commit 0613c7f
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 0613c7f

Please sign in to comment.