Browse Source

Rename dataloader to add support for valid loader

Lengyue 2 years ago
parent
commit
b6b478701a
2 changed files with 2 additions and 2 deletions
  1. 1 1
      speech_lm/configs/hubert_vq.yaml
  2. 1 1
      speech_lm/configs/pretrain.yaml

+ 1 - 1
speech_lm/configs/hubert_vq.yaml

@@ -43,7 +43,7 @@ dataset:
   _target_: speech_lm.datasets.hubert_vq.HubertVQDataset
   filelist: libritts-r.filelist
 
-dataloader:
+train_dataloader:
   _target_: torch.utils.data.DataLoader
   dataset: ${dataset}
   batch_size: ${schedule.micro_batch_size}

+ 1 - 1
speech_lm/configs/pretrain.yaml

@@ -78,7 +78,7 @@ dataset:
   probabilities: [0.4, 0.3, 0.3]
   seed: 42
 
-dataloader:
+train_dataloader:
   _target_: torch.utils.data.DataLoader
   dataset: ${dataset}
   batch_size: ${schedule.micro_batch_size}