From d82405cd4842b2a82a0834bbc05867e1a0c97cf2 Mon Sep 17 00:00:00 2001 From: Nian Date: Wed, 8 May 2024 17:27:11 +0800 Subject: [PATCH] Fix validation dataset definition, removed unused `test_dataset` Explaination see in: https://github.com/Audio-WestlakeU/ATST-SED/issues/11 --- train/train_stage2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train/train_stage2.py b/train/train_stage2.py index 01ad4c8..95491c7 100644 --- a/train/train_stage2.py +++ b/train/train_stage2.py @@ -210,7 +210,7 @@ def single_run( batch_sizes = config["training"]["batch_size"] samplers = [torch.utils.data.RandomSampler(x) for x in tot_train_data] batch_sampler = ConcatDatasetBatchSampler(samplers, batch_sizes) - valid_dataset = torch.utils.data.ConcatDataset([weak_val, synth_val, strong_val, test_dataset]) + valid_dataset = torch.utils.data.ConcatDataset([weak_val, synth_val, strong_val]) ##### training params and optimizers ############ epoch_len = min(