From 230ee5c99129d79e3260e37a425f8e93447e2cb8 Mon Sep 17 00:00:00 2001 From: pkufool Date: Thu, 16 Jun 2022 15:27:47 +0800 Subject: [PATCH] Disable drop_last in testing time --- .../ASR/pruned_transducer_stateless3/asr_datamodule.py | 1 - 1 file changed, 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py b/egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py index b54d1aa39..1df7f9ee5 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py @@ -291,7 +291,6 @@ class AsrDataModule: max_duration=self.args.max_duration, shuffle=False, num_buckets=self.args.num_buckets, - drop_last=True, ) logging.debug("About to create test dataloader") test_dl = DataLoader(