fix sortagrad, test=asr

PaddlePaddle · Feb 15, 2022 · 95d5274 · 95d5274
1 parent 7f970bb
commit 95d5274
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/paddlespeech/s2t/io/batchfy.py b/paddlespeech/s2t/io/batchfy.py
@@ -419,7 +419,7 @@ def make_batchset(
  # sort it by input lengths (long to short)
  sorted_data = sorted(
  d.items(),
- key=lambda data: int(data[1][batch_sort_key][batch_sort_axis]["shape"][0]),
+ key=lambda data: float(data[1][batch_sort_key][batch_sort_axis]["shape"][0]),
  reverse=not shortest_first, )
  logger.info("# utts: " + str(len(sorted_data)))
 

diff --git a/paddlespeech/s2t/io/dataloader.py b/paddlespeech/s2t/io/dataloader.py
@@ -61,7 +61,7 @@ class BatchDataLoader():
  def __init__(self,
  json_file: str,
  train_mode: bool,
- sortagrad: bool=False,
+ sortagrad: int=0,
  batch_size: int=0,
  maxlen_in: float=float('inf'),
  maxlen_out: float=float('inf'),