mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 10:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fit type hint

Browse Source
This commit is contained in:
yfy62 2023-04-26 17:45:00 +08:00
parent 23a9b66295
commit 5e7e1a350e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/gigaspeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -564,7 +564,7 @@ def save_checkpoint(
def compute_loss(
params: AttributeDict,
model: nn.Module,
model: Union[nn.Module, DDP],
sp: spm.SentencePieceProcessor,
pl: UniqLexicon,
batch: dict,
@ -948,7 +948,7 @@ def run(rank, world_size, args):
valid_cuts = gigaspeech.dev_cuts()
valid_dl = gigaspeech.valid_dataloaders(valid_cuts)
if 0 and not params.print_diagnostics:
if not params.print_diagnostics:
scan_pessimistic_batches_for_oom(
model=model,
train_dl=train_dl,