Fix validation dataset definition, removed unused test_dataset

Explaination see in: #11
Audio-WestlakeU · May 8, 2024 · d82405c · d82405c
1 parent 0ac8073
commit d82405c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(