enable shuffle in dataloader.

Author: csukuangfj

egs/aishell/s10/chain/egs_dataset.py

@@ -17,7 +17,10 @@
 from common import splice_feats
 
 
-def get_egs_dataloader(egs_dir, egs_left_context, egs_right_context):
+def get_egs_dataloader(egs_dir,
+                       egs_left_context,
+                       egs_right_context,
+                       shuffle=True):
 
     dataset = NnetChainExampleDataset(egs_dir=egs_dir)
     frame_subsampling_factor = 3
@@ -32,6 +35,7 @@ def get_egs_dataloader(egs_dir, egs_left_context, egs_right_context):
 
     dataloader = DataLoader(dataset,
                             batch_size=batch_size,
+                            shuffle=shuffle,
                             num_workers=0,
                             collate_fn=collate_fn)
     return dataloader

egs/aishell/s10/chain/train.py

@@ -181,7 +181,8 @@ def main():
 
     dataloader = get_egs_dataloader(egs_dir=args.cegs_dir,
                                     egs_left_context=args.egs_left_context,
-                                    egs_right_context=args.egs_right_context)
+                                    egs_right_context=args.egs_right_context,
+                                    shuffle=True)
 
     optimizer = optim.Adam(model.parameters(),
                            lr=learning_rate,