mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

nevermind it breaks training

Browse Source
This commit is contained in:
mrq 2024-05-25 18:03:43 -05:00
parent 85f9684720
commit 05cd8b797e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/train.py
View File

@ -155,9 +155,11 @@ def run_eval(engines, eval_name, dl):
def train(): def train():
"""
parser = argparse.ArgumentParser("VALL-E TTS") parser = argparse.ArgumentParser("VALL-E TTS")
parser.add_argument("--eval", action="store_true") parser.add_argument("--eval", action="store_true")
args = parser.parse_args() args = parser.parse_args()
"""
setup_logging(cfg.log_dir) setup_logging(cfg.log_dir)
@ -180,8 +182,10 @@ def train():
qnt.unload_model() qnt.unload_model()
"""
if args.eval: if args.eval:
return eval_fn(engines=trainer.load_engines()) return eval_fn(engines=trainer.load_engines())
"""
""" """
if cfg.trainer.load_webui: if cfg.trainer.load_webui: