from local

This commit is contained in:
dohe0342 2023-02-02 15:27:55 +09:00
parent 22a125dfa4
commit f8771a569f
2 changed files with 2 additions and 1 deletions

View File

@ -643,7 +643,8 @@ def run(rank, world_size, args):
optimizer.load_state_dict(checkpoints["optimizer"]) optimizer.load_state_dict(checkpoints["optimizer"])
aishell = AishellAsrDataModule(args) aishell = AishellAsrDataModule(args)
train_dl = aishell.train_dataloaders(aishell.train_cuts()) #train_dl = aishell.train_dataloaders(aishell.train_cuts())
train_dl = aishell.valid_dataloaders(aishell.valid_cuts())
valid_dl = aishell.valid_dataloaders(aishell.valid_cuts()) valid_dl = aishell.valid_dataloaders(aishell.valid_cuts())
for epoch in range(params.start_epoch, params.num_epochs): for epoch in range(params.start_epoch, params.num_epochs):