remove quantity limit in valid dataset

ZhangXInFD · Jun 9, 2024 · 30c96fb · 30c96fb
1 parent cd8522b
commit 30c96fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/speechtokenizer/trainer/trainer.py b/speechtokenizer/trainer/trainer.py
@@ -147,7 +147,7 @@ def __init__(
                                 segment_size=segment_size,
                                 downsample_rate=generator.downsample_rate,
                                 sample_rate=self.sample_rate)
-        self.valid_ds = audioDataset(file_list=valid_file_list[:1000],
+        self.valid_ds = audioDataset(file_list=valid_file_list,
                                     segment_size=self.sample_rate * 30,
                                     downsample_rate=generator.downsample_rate,
                                     sample_rate=self.sample_rate,