|
|
@ -254,12 +254,12 @@ class U2Trainer(Trainer):
|
|
|
|
config.data.manifest = config.data.test_manifest
|
|
|
|
config.data.manifest = config.data.test_manifest
|
|
|
|
config.data.keep_transcription_text = True
|
|
|
|
config.data.keep_transcription_text = True
|
|
|
|
config.data.augmentation_config = ""
|
|
|
|
config.data.augmentation_config = ""
|
|
|
|
config.data.min_input_len = 0.0 # second
|
|
|
|
# config.data.min_input_len = 0.0 # second
|
|
|
|
config.data.max_input_len = float('inf') # second
|
|
|
|
# config.data.max_input_len = float('inf') # second
|
|
|
|
config.data.min_output_len = 0.0 # tokens
|
|
|
|
# config.data.min_output_len = 0.0 # tokens
|
|
|
|
config.data.max_output_len = float('inf') # tokens
|
|
|
|
# config.data.max_output_len = float('inf') # tokens
|
|
|
|
config.data.min_output_input_ratio = 0.00
|
|
|
|
# config.data.min_output_input_ratio = 0.00
|
|
|
|
config.data.max_output_input_ratio = float('inf')
|
|
|
|
# config.data.max_output_input_ratio = float('inf')
|
|
|
|
test_dataset = ManifestDataset.from_config(config)
|
|
|
|
test_dataset = ManifestDataset.from_config(config)
|
|
|
|
# return text ord id
|
|
|
|
# return text ord id
|
|
|
|
self.test_loader = DataLoader(
|
|
|
|
self.test_loader = DataLoader(
|
|
|
|