mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 10:16:14 +00:00
Merge branch 'k2-fsa:master' into fix/k2ssl-multi-gpu
This commit is contained in:
commit
967bf92d87
@ -181,7 +181,7 @@ class YesNoAsrDataModule(DataModule):
|
||||
train = K2SpeechRecognitionDataset(
|
||||
cut_transforms=transforms,
|
||||
input_strategy=OnTheFlyFeatures(
|
||||
FbankConfig(sampling_rate=8000, num_mel_bins=23)
|
||||
Fbank(FbankConfig(sampling_rate=8000, num_mel_bins=23))
|
||||
),
|
||||
return_cuts=self.args.return_cuts,
|
||||
)
|
||||
@ -222,9 +222,11 @@ class YesNoAsrDataModule(DataModule):
|
||||
|
||||
logging.debug("About to create test dataset")
|
||||
test = K2SpeechRecognitionDataset(
|
||||
input_strategy=OnTheFlyFeatures(Fbank(FbankConfig(num_mel_bins=23)))
|
||||
input_strategy=(
|
||||
OnTheFlyFeatures(Fbank(FbankConfig(num_mel_bins=23)))
|
||||
if self.args.on_the_fly_feats
|
||||
else PrecomputedFeatures(),
|
||||
else PrecomputedFeatures()
|
||||
),
|
||||
return_cuts=self.args.return_cuts,
|
||||
)
|
||||
sampler = DynamicBucketingSampler(
|
||||
|
@ -110,13 +110,6 @@ def str2bool(v):
|
||||
raise argparse.ArgumentTypeError("Boolean value expected.")
|
||||
|
||||
|
||||
def clear_log_handlers():
|
||||
logger = logging.getLogger()
|
||||
handlers = logger.handlers[:]
|
||||
for handler in handlers:
|
||||
logger.removeHandler(handler)
|
||||
|
||||
|
||||
def setup_logger(
|
||||
log_filename: Pathlike,
|
||||
log_level: str = "info",
|
||||
@ -133,8 +126,6 @@ def setup_logger(
|
||||
use_console:
|
||||
True to also print logs to console.
|
||||
"""
|
||||
clear_log_handlers()
|
||||
|
||||
now = datetime.now()
|
||||
date_time = now.strftime("%Y-%m-%d-%H-%M-%S")
|
||||
if dist.is_available() and dist.is_initialized():
|
||||
|
Loading…
x
Reference in New Issue
Block a user