From a85312c0ebdf0fbbda29868baddc10ef6e25760a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Sun, 24 Mar 2024 15:29:14 +0800 Subject: [PATCH] finetune --- funasr/datasets/dataloader_entry.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/datasets/dataloader_entry.py b/funasr/datasets/dataloader_entry.py index 2222efbf2..abb28285a 100644 --- a/funasr/datasets/dataloader_entry.py +++ b/funasr/datasets/dataloader_entry.py @@ -50,7 +50,7 @@ class DataloaderMapStyle: batch_sampler_val = batch_sampler_class(self.dataset_val, is_training=False, **self.kwargs.get("dataset_conf")) batch_sampler["batch_sampler"].set_epoch(epoch) - batch_sampler_val.set_epoch(epohc) + batch_sampler_val["batch_sampler"].set_epoch(epoch) dataloader_tr = torch.utils.data.DataLoader(self.dataset_tr, collate_fn=self.dataset_tr.collator, **batch_sampler) dataloader_val = torch.utils.data.DataLoader(self.dataset_val, collate_fn=self.dataset_val.collator, **batch_sampler_val)