fix dist_sampler AttributeError

3 years ago · 4ad580406a
parent c75906462e
commit 4ad580406a
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)