From 3cf194c22266fe89f7b474438bc2a1cba4a49c08 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 14:06:04 +0900 Subject: [PATCH] from local --- .../.train_adapter.py.swp | Bin 77824 -> 77824 bytes .../train_adapter.py | 4 +--- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp index f92236046dbaf70c8a246c6888bdfb0ba4223e4b..4c80a5b9d6953a29b5fef6e094eed8b2cd3f924c 100644 GIT binary patch delta 523 zcmXZZPbkA-9LMp`Y&J80(=L-52XVmx;iPa74qE=i!QTT($$^_mxS{MYg^O}D)=$PR z{vJfQxJ)S}!ohMIO5QV{dOgp#=XvTme5M~P(+?JBe_4BDi@Um{Oe9O%c`F}`1)Zze zrU4_E?Fo2uIX^WeE(hX|NGMa zk&llg=@FqimU(od0~L6tjwhTUUQPI4q8!Qnm-rUvme2H^<7?i3F}6=!cHce{XG=<<+NN@4dt8J>RX)cdN7Z*qsj1gi2nVNvv72BmE$p&rDF^vZHHPx+4N)#sB_t=49< O&4<%97WL@7#{CQ6>s`?R diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py index 902e67036..ffab01c73 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py @@ -1363,8 +1363,6 @@ def run(rank, world_size, args, wb=None): scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs) - ''' - if checkpoints and ("optimizer" in checkpoints or "optimizer_enc" in checkpoints): if params.multi_optim: logging.info("Loading optimizer state dict") @@ -1387,7 +1385,7 @@ def run(rank, world_size, args, wb=None): else: logging.info("Loading scheduler state dict") scheduler.load_state_dict(checkpoints["scheduler"]) - ''' + if params.print_diagnostics: opts = diagnostics.TensorDiagnosticOptions( 2**22