From c0fff54518e44e366a3ea382196aee1f123deb7c Mon Sep 17 00:00:00 2001 From: enhuiz Date: Thu, 12 Jan 2023 20:30:59 +0800 Subject: [PATCH] Rename --- vall_e/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vall_e/train.py b/vall_e/train.py index 5dbcd06..b5e0b37 100644 --- a/vall_e/train.py +++ b/vall_e/train.py @@ -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)