From 4d458a00828643bbec7937bd3437cc0b94c83049 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 26 Jan 2023 14:39:31 +0900
Subject: [PATCH] from local

---
 .../.train_adapter.py.swp                     | Bin 90112 -> 90112 bytes
 .../train_adapter.py                          |  25 +++++++++---------
 2 files changed, 13 insertions(+), 12 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 5a4b51556d01f5b2386be2d7006a205b94f53680..c6090be6d9f55007b42d9a4e0b3a90182789d1ce 100644
GIT binary patch
delta 425
zcmXxgPbh<N9LMqRvuBSXTeg`UT$Yr7qg_{aa`BK5InKY7B4I1GrLD}B^c<wNlN`9|
zDU-C;4kW~siySDNMeX2#cge3_^{MZ-Z{P1PRnSre%`+Jo)%$(H&VcZ0MZTSPQ|*VI
z2UQXCT^)BQ33JG0hU_Pli<ZBhgy9^#Os1<v;)r1ct?<Huk1COS9AO7h44@5y=Sq<?
z93zJ;GFZn1dZD8Sel$SvXcH-6(<VvTAf&MjKkWFaU@%Uxhb-cVp&M>|YAlEg9AF<2
zETIc^c(aPU;stqRu!uf1qsH38A%v1Da*I6XF@r{w`KucgEnB6+wlAz$P51DZX3SpI
p8(H0(d+Jbj71zStSaf<N8VUUezqzVMRO7nbv=$8Kf??%c`2}#OR2%>R

delta 418
zcmXBQ!7GCS7{~GF?R{U&vUzEyn5Nc3S;J{X$i>PEwUw>baNy>^Ws|m&;-FS<%*7wz
zqSl)n6gz0E$zeO<<}@ccObPjxr#|&N{CawNawR=i(p_u8SUBPh1%kqKihLXObz9N(
zq$*MCT}?TZjJ4EdO?B865vTg!$(Y9CL-wFuWEUAkF@j#W@oN)#MHLq~#0F;JhZ8z9
zytRomP{$<-IK>Il*hB=w7{E)b$aAY?rAD|%8gVS38#a8juoxATkw*$Mm_`pib&)1Y
zxWf@vF@bT6!H$L|a*H$UU<EPF$29~G?6~1KSJ=WF`r%ZIHA5+QlZs}Qm%ns#XwqSx
phux+p(Pt*A{=#TL88naQf6RQ`u=c{+s`=RT?T3=rp`=n#{s5C;Ou+yE

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 6e49b2df3..7c1cfadcf 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
@@ -1647,23 +1647,24 @@ def run_adapter(rank, world_size, args, wb=None):
             world_size=world_size,
             rank=rank,
             wb=wb,
-            pl_texts=pl_texts
+            pl_texts=pl_texts,
         )
 
         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!")