fix some bug, test=asr

pull/1577/head
huangyuxin 4 years ago
parent 6da8465f14
commit e1b581b622

@ -85,7 +85,7 @@ scheduler: warmuplr
scheduler_conf: scheduler_conf:
warmup_steps: 25000 warmup_steps: 25000
lr_decay: 1.0 lr_decay: 1.0
log_interval: 1 log_interval: 100
checkpoint: checkpoint:
kbest_n: 50 kbest_n: 50
latest_n: 5 latest_n: 5

@ -239,7 +239,7 @@ class U2Trainer(Trainer):
n_iter_processes=config.num_workers, n_iter_processes=config.num_workers,
subsampling_factor=1, subsampling_factor=1,
num_encs=1, num_encs=1,
dist_sampler=True, dist_sampler=config.get('dist_sampler', False),
shortest_first=False) shortest_first=False)
self.valid_loader = BatchDataLoader( self.valid_loader = BatchDataLoader(
@ -260,7 +260,7 @@ class U2Trainer(Trainer):
n_iter_processes=config.num_workers, n_iter_processes=config.num_workers,
subsampling_factor=1, subsampling_factor=1,
num_encs=1, num_encs=1,
dist_sampler=True, dist_sampler=config.get('dist_sampler', False),
shortest_first=False) shortest_first=False)
logger.info("Setup train/valid Dataloader!") logger.info("Setup train/valid Dataloader!")
else: else:

@ -160,9 +160,12 @@ class DefaultInitializerContext(object):
self.init_type = init_type self.init_type = init_type
def __enter__(self): def __enter__(self):
if self.init_type is None:
return
else:
from paddlespeech.s2t.modules import align from paddlespeech.s2t.modules import align
align.global_init_type = self.init_type align.global_init_type = self.init_type
return self return
def __exit__(self, exc_type, exc_val, exc_tb): def __exit__(self, exc_type, exc_val, exc_tb):
from paddlespeech.s2t.modules import align from paddlespeech.s2t.modules import align

Loading…
Cancel
Save