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

Add mix precision support

Browse Source
This commit is contained in:
pkufool 2022-04-11 15:27:24 +08:00 committed by Daniel Povey
parent 8cb727e24a
commit cc0d4ffa4f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -870,7 +870,7 @@ def run(rank, world_size, args):
valid_cuts += librispeech.dev_other_cuts()
valid_dl = librispeech.valid_dataloaders(valid_cuts)
if not params.print_diagnostics:
if not params.print_diagnostics and not params.use_fp16:
scan_pessimistic_batches_for_oom(
model=model,
train_dl=train_dl,