From 5379c8e9fa13f6f2364b4a0db89fa3074266fb58 Mon Sep 17 00:00:00 2001 From: Wei Kang Date: Thu, 16 Jun 2022 15:43:48 +0800 Subject: [PATCH] Disable drop_last in testing time (#427) --- .../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(