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

Rename

Browse Source
This commit is contained in:
enhuiz 2023-01-12 20:30:59 +08:00
parent 85c6a11f26
commit c0fff54518
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/train.py
View File

@ -30,7 +30,7 @@ def load_engines():
def main():
setup_logging(cfg.log_dir)
train_dl, train200_dl, val_dl, test_dl = create_train_val_dataloader()
train_dl, train_for_val_dl, val_dl, test_dl = create_train_val_dataloader()
def train_feeder(engines, batch, name):
model = engines["model"]
@ -112,7 +112,7 @@ def main():
_logger.info(f"{json.dumps(stats)}.")
def eval_fn(engines):
run_eval(engines, "train200", train200_dl)
run_eval(engines, "train_for_val", train_for_val_dl)
run_eval(engines, "val", val_dl)
run_eval(engines, "test", test_dl)