From 000dc1af5cbb040ed3bef097f40b5fe15939a812 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 18 Apr 2023 14:08:23 +0900
Subject: [PATCH] from local

---
 egs/librispeech/ASR/.full_ft.sh.swp           | Bin 12288 -> 12288 bytes
 .../.full_ft.py.swp                           | Bin 98304 -> 98304 bytes
 .../full_ft.py                                |   9 ++-------
 3 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/egs/librispeech/ASR/.full_ft.sh.swp b/egs/librispeech/ASR/.full_ft.sh.swp
index edbdcbcd2c16891ce99695d5207ac19631f6982f..c737dfb15507d59577d6d882b1b59ddbf1eb2733 100644
GIT binary patch
delta 31
lcmZojXh;xCG6?hZRj|}EU;qLE28M;Ib}36IY!rK?4*-C=2$=u?

delta 31
lcmZojXh;xCG6?hZRj|}EU;qLE28RDCb}4QB8^vDf0|0}b2-W}q

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp
index e454939ab07552211496931dc9e1fe29cb5f6639..cfbbff9c764b66b3edfc85dd98b1a52f0fe22773 100644
GIT binary patch
delta 509
zcmYk&O(?^07{KvoUbY!~`7>or*-<Vt<%m)u2hk2Y$XjhKneDJznX-Sj(@NR@e{oP8
zl@xZ6gL06Bl=5;>n>dimD2dB_my2J0emy;JheuV7s%qRVDTZMFU1?gfmPKT{iq9bu
z)Rk3l>dGc^h&23gqZv+=ts*Zt#Uc_IhaEP2HHZ{Zz!~<ig(L!SqEs((i7jL>4Iiv{
zwMa@{EFve^K^C*<LpK!s)QQ~V04qpi89o%v%pr*d#4v_V6igzwxWO9y7)A#ycr%Jz
z7*ita1R8P}!XQeuBG*{MD9kY7yGG;~DpHukBzj@TT{TOQML#~c^)o6r&me#Sxbb8V
zdC-q?U3zQWs%dSyBks~PrP1^6MPlLbZ1k_<)E{&0x)N{M*FNiAm8SVvFcg`ZNrXK;
Uo{?BIB$cGTc|MW%Zd@0C0qOr>xBvhE

delta 449
zcmWmBPbh-{9LDiy%)5W4y>{FV4%04c3vu8`ilye{=BQRyT&e9Xaw6tUYAGi<xX2q0
z{!FRW4innJ!NFmr6`91z!PnvQ^t(Mh{VdC{EaSLky0Xrv8O{gel3razNXpvgR;uvW
zC6%y%S@hv2D1D=hP0U~#8H^x?Hk@}#C)h<AlNdl05xDUlkQz9HjSLd-qTyFTH-2dk
zIZU7zF$hhcRKWq3v4|-QA&f7tR7Dxr$RUcRheDj<2pgD31SUeL8;mmw(i+Dc#u0}H
zPaV=7YFNbz5(uL1mew(bQ4Hd@UAn{t4pG7q(&&Z<l{P7lBzo}whiCh$*l)LzZu_H{
Wu<v(#&f9h^;M`V+T~2EM;pGn#h)d4^

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py
index 7522912d2..070d8349f 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py
@@ -1262,13 +1262,8 @@ def run(rank, world_size, args, wb=None):
 
     num_param = sum([p.numel() for p in model.parameters()])
     logging.info(f"Number of model parameters: {num_param}")
-
-    assert params.save_every_n >= params.average_period
-    model_avg: Optional[nn.Module] = None
-    if rank == 0:
-        # model_avg is only used with rank 0
-        model_avg = copy.deepcopy(model).to(torch.float64)
-
+    
+    model_avg = None
     assert params.start_epoch > 0, params.start_epoch
     checkpoints = load_checkpoint_if_available(
         params=params, model=model, model_avg=model_avg