pull/768/head
Hui Zhang 4 years ago
parent c484d537c2
commit d64cdc7838

@ -233,7 +233,7 @@ class U2Trainer(Trainer):
batch_frames_inout=0, batch_frames_inout=0,
preprocess_conf=config.collator.augmentation_config, preprocess_conf=config.collator.augmentation_config,
n_iter_processes=config.collator.num_workers, n_iter_processes=config.collator.num_workers,
subsampling_factor=0, subsampling_factor=1,
num_encs=1) num_encs=1)
self.valid_loader = BatchDataLoader( self.valid_loader = BatchDataLoader(

@ -65,7 +65,6 @@ class SpecAugmentor(AugmentorBase):
self.n_time_masks = n_time_masks self.n_time_masks = n_time_masks
self.p = p self.p = p
# adaptive SpecAugment # adaptive SpecAugment
self.adaptive_number_ratio = adaptive_number_ratio self.adaptive_number_ratio = adaptive_number_ratio
self.adaptive_size_ratio = adaptive_size_ratio self.adaptive_size_ratio = adaptive_size_ratio

Loading…
Cancel
Save