From 87b37383ab8f7db5671f112a95091cdcef56b124 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Wed, 25 Jan 2023 16:51:48 +0900
Subject: [PATCH] from local

---
 .../.train_adapter.py.swp                     | Bin 81920 -> 86016 bytes
 .../train_adapter.py                          |  23 +++++++++---------
 2 files changed, 12 insertions(+), 11 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 7ec946466e6b7132721add1bebd25bd4eb8bc251..83aa46bd6a47f703eb10cc9e4106cb3c5e6b2aa5 100644
GIT binary patch
delta 1089
zcmXxiYe*DP7=Yn#)OFRpx#lI)V6rUxk*?h+MM|U6NK-=%BcilY5|zNniq=Yrl7bv)
z5rs0yvMA7nGSV!vKtgFN>;@W?fwqw)Nou!u(GEN_!x_#n@P1WQx~faM(A~D&v~+cK
ztW8*fBJT#C734o`4Ru?TSx)c3UPUo#E^C-I*z=xMGJTUy=Nqfu+@|`drh2o;5Io0?
znssfQU9n8>xY%un0|$|VI0WOjNn{M)@fxr20F9_YHBKT6X;_107&r1wRG<I}h{aEX
z$SC>^VwZLv?%)FK*nsu0!W|&;8QrKv8FG<><p{zje|`pCsK*f;#zutWiyz;?3p~d)
z*l`St{OlqldXZtY;uI>73KNXz_Z4YIDT=TU*+@h<`g9^aXhQ?a5rscKB5!dKM^TDR
zSO^1#y*amnO9RehDdO?Xi$BIa+{P_5AQKxfsW``z$BdEc%qrfi?kO)TC{m-<c$=D(
zq}me1(X>Pf(fW!<E@zI~cNK~#ibyYqJ>^Sg2IB+myKxmKumh`L##g$(#1lNm4P1wY
zGHgR0a<KrB2!bzqXkUv|tTfTs#DjvT#D9bq)Zz?^k&ST7#z(q$;XWMLj1(+J4D=YH
z`(0?LLkTjFjyZ7C{r^lf`_uRe8=k?)MiwTC--!-XBOj@Vz<Zhxq6L*mMj{dr4Kv(y
zZ$cw#;F*ne_(69UZsGv8AUTc8Tucys03B$AXCMlpgAclC--UBHjpHcAe1xOdtJS4f
zG*g?O(>G<ppzZ$^s+lsvH22>Ct$cfkmN$7w8_d=@c19_Ynq_KU-PQ>0?3BeTM*aa|
C0=8cO

delta 536
zcmXBQO(?^07{Kx8ZyOt%y_9iT?c_kQ$lC>R+L#=WxhM`2C)Ao^ivx1fav^PTlLK2Q
zR)~Yk@=_?YoR=}>Ewp^)SD#<cQ_u7C^aR41KtN;db96b~_PR2M@Ju4ziSz#6!!7Hh
zQL#i-rnsVbgBRw)sMhjnSa`5U>}tCqo6?N(U+)c?8_WGVqewbm<O(q?V;mk-q6i8;
z3?lEi#~n@(#vEobgHE{7j2e9DnF${TP>%}y<cYlEHc$K#BiP3hrs0AU1xV+LJmVA-
z7)1{}sD%|t9n0VtEAV3yEwJH5%N*R`8XK6#6iT&zk*6Gy6t*!BA6j9C3AY?=9U~aV
z5W3I+8!j~>aqJ+3F%%;U3AM-?0vJIXN?=4vMPC(P2#cu3U;dWG#yG$pA_$=aE@YJ5
Yqb9|m4}QCo!LuJ@baPOvthT3r0q$2{wg3PC

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 15d8222f1..e21c475dc 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
@@ -1464,17 +1464,18 @@ def run(rank, world_size, args, wb=None):
         if params.print_diagnostics:
             diagnostic.print_diagnostics()
             break
-
-        save_checkpoint(
-            params=params,
-            model=model,
-            model_avg=model_avg,
-            optimizer=optimizer,
-            scheduler=scheduler,
-            sampler=train_dl.sampler,
-            scaler=scaler,
-            rank=rank,
-        )
+        
+        if epoch % 10 == 0:
+            save_checkpoint(
+                params=params,
+                model=model,
+                model_avg=model_avg,
+                optimizer=optimizer,
+                scheduler=scheduler,
+                sampler=train_dl.sampler,
+                scaler=scaler,
+                rank=rank,
+            )
 
     logging.info("Done!")