fix dist_sampler AttributeError (#2918)

2 years ago · 17a7ebddfa
parent acfa057dc7
commit 17a7ebddfa
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/s2t/io/dataloader.py
+++ b/paddlespeech/s2t/io/dataloader.py
@ -464,5 +464,5 @@ class DataLoaderFactory():
                subsampling_factor=config.subsampling_factor,
                load_aux_output=config.get('load_transcript', None),
                num_encs=config.num_encs,
-                dist_sampler=config.dist_sampler,
+                dist_sampler=config.get('dist_sampler', None),
                shortest_first=config.shortest_first)