anhnv125 commited on
Commit
3ebc8ea
1 Parent(s): c729190

Update dataset.py

Browse files
Files changed (1) hide show
  1. dataset.py +1 -1
dataset.py CHANGED
@@ -140,7 +140,7 @@ class TestLoader(Dataset):
140
 
141
  class BlindTestLoader(Dataset):
142
  def __init__(self, test_dir):
143
- self.data_list = glob.glob(os.path.join(test_dir, '*.wav'))[:10]
144
  self.sr = CONFIG.DATA.sr
145
  self.stride = CONFIG.DATA.stride
146
  self.chunk_len = CONFIG.DATA.window_size
 
140
 
141
  class BlindTestLoader(Dataset):
142
  def __init__(self, test_dir):
143
+ self.data_list = glob.glob(os.path.join(test_dir, '*.wav'))
144
  self.sr = CONFIG.DATA.sr
145
  self.stride = CONFIG.DATA.stride
146
  self.chunk_len = CONFIG.DATA.window_size