From df419089c85c438dacf90462ac17bc2625811beb Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 10 Dec 2022 14:09:26 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 102400 -> 102400 bytes .../train.py | 10 ++++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp index 2836b13fe87949c9c92cc25045a8b9346acf2871..94a9cb55dd909678bdd6ef69b4079a8ea52aff00 100644 GIT binary patch delta 338 zcmXBP&nts*9Ki9);|& zb8yny!Bsmc{0aC2T*Of;@8w&s`u6$N=TogY)tcjNWY(5f)ZA1?c&GF@_YrsDH*K4pBf3^O(ReIy2*lq17c)Mu-xU z5WLwUkGMhwOISqKmIaCP`DUjNJfVRzoT7vrvdF+i44oT#;O`feGB@{L(@)hS)~I>a zNpq{y&FJ`ul_-Y(=6bN54|d97p|oeNKRvUmhipYpU8lPzWvdeIDpS$Zrj_^1i&n>N K{m|ByUHAjZ{6Gf) delta 332 zcmXBOze_@47{&4D_3A~6P>E|x=~eQjK|+F%+M+4Y5Q5gY{(vOOpdyDLA+Yv0u+bq1 z+UyeT4Glt6e?SqfO;KA8eWM3H@Nmv@+C977vz>BoeWmCw%;toPiVVijDwBQZC!$(@ zEcz80E0Grh|0R)|S4R)FNC!16B8e${ghcLPaDXaG$RmRZgkd8DKPYmIKFU~z3kz?S z$PG?$f(}+OZ^@FRc}DTgt|tu8$2mIK#TttE+pdv97~X!v?LHUPT=%h{V&?NI?kCb0 z0e>WuQ0A`Y7*9Lquo_oc^P*jIsZ-{Cdd9r!WVF%rsvDKfT4igeS>N8v9uM{HP^)15 E7j`2+`2YX_ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index d84188963..cc97dd06f 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -74,11 +74,12 @@ from zipformer import Zipformer from data2vec_encoder import FairSeqData2VecEncoder from icefall import diagnostics -from icefall.checkpoint import load_checkpoint, remove_checkpoints -from icefall.checkpoint import save_checkpoint as save_checkpoint_impl -from icefall.checkpoint import ( +from icefall.checkpoint import remove_checkpoints +from icefall.checkpoint import update_averaged_model +from checkpoint import ( + save_checkpoint as save_checkpoint_impl, save_checkpoint_with_global_batch_idx, - update_averaged_model, + load_checkpoint ) from icefall.dist import cleanup_dist, setup_dist from icefall.env import get_env_info @@ -92,6 +93,7 @@ from icefall.utils import ( save_args, ) +#from icefall.checkpoint import save_checkpoint as save_checkpoint_impl LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler]