diff --git a/deepspeech/exps/deepspeech2/model.py b/deepspeech/exps/deepspeech2/model.py index a163ef4ca133465f7d93478f1bda48c07fca34fa..b1ff5c591404152f7c96357e564ca700a2d73c74 100644 --- a/deepspeech/exps/deepspeech2/model.py +++ b/deepspeech/exps/deepspeech2/model.py @@ -323,12 +323,12 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): config.data.manifest = config.data.test_manifest config.data.keep_transcription_text = True config.data.augmentation_config = "" - config.data.min_input_len = 0.0 # second - config.data.max_input_len = float('inf') # second - config.data.min_output_len = 0.0 # tokens - config.data.max_output_len = float('inf') # tokens - config.data.min_output_input_ratio = 0.00 - config.data.max_output_input_ratio = float('inf') + # config.data.min_input_len = 0.0 # second + # config.data.max_input_len = float('inf') # second + # config.data.min_output_len = 0.0 # tokens + # config.data.max_output_len = float('inf') # tokens + # config.data.min_output_input_ratio = 0.00 + # config.data.max_output_input_ratio = float('inf') test_dataset = ManifestDataset.from_config(config) # return text ord id diff --git a/deepspeech/exps/u2/model.py b/deepspeech/exps/u2/model.py index 7a409d98807c18154e3170e2974e4bb5e3e601d3..0bcd1d22421b01d2f9df6e7a208b0557256465ac 100644 --- a/deepspeech/exps/u2/model.py +++ b/deepspeech/exps/u2/model.py @@ -254,12 +254,12 @@ class U2Trainer(Trainer): config.data.manifest = config.data.test_manifest config.data.keep_transcription_text = True config.data.augmentation_config = "" - config.data.min_input_len = 0.0 # second - config.data.max_input_len = float('inf') # second - config.data.min_output_len = 0.0 # tokens - config.data.max_output_len = float('inf') # tokens - config.data.min_output_input_ratio = 0.00 - config.data.max_output_input_ratio = float('inf') + # config.data.min_input_len = 0.0 # second + # config.data.max_input_len = float('inf') # second + # config.data.min_output_len = 0.0 # tokens + # config.data.max_output_len = float('inf') # tokens + # config.data.min_output_input_ratio = 0.00 + # config.data.max_output_input_ratio = float('inf') test_dataset = ManifestDataset.from_config(config) # return text ord id self.test_loader = DataLoader( diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py index 220914f3f16e7da9fcdbe821dbf9f5367570d2ec..e630febbc7862b037a6b933fb73511831766b7fc 100644 --- a/deepspeech/training/trainer.py +++ b/deepspeech/training/trainer.py @@ -264,6 +264,7 @@ class Trainer(): @mp_tools.rank_zero_only def destory(self): + """Close visualizer to avoid hanging after training""" # https://github.com/pytorch/fairseq/issues/2357 if self.visualizer: self.visualizer.close()