From fecb601ef14e0e700a6544c9b5e74c2a4fd9f074 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 19:01:59 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 81920 -> 81920 bytes .../train_lora.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp index 6d8ba42ea594734aec7db08fdb1ec459d6167f25..13b3abd8731e923e8f4d9d3cb1c6b6d0bd60fbf2 100644 GIT binary patch delta 86 zcmZo@U~On%6-_b-^Ym4)&@*HJ0s#hw;|BRDPa8$IiZZ_EXWhiZz#uoBQ-Cp=QEq#x f0OL*`Nd+X3nWj*bn3t_!YpY;jwcT8n@uU&}j1d&T delta 76 zcmZo@U~On%6-_b-^Ym4)&@*HJ0s#hwi3a&8R~kjPiZZ_EXFb5fz@RgoQ-Cp=QD=Lq f0OL*`cD>yEl+v8k$q)5}C-3>iyWLfmakmlx`!5y| diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py index edf8e8249..d90de5f94 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py @@ -1605,7 +1605,7 @@ def run_adapter(rank, world_size, args, wb=None): lora_modules = [] for modules in model.modules(): if isinstance(modules, fairseq.modules.multihead_attention.MultiheadAttention): - logging.info(modules) + if rank == 0: logging.info(modules) ''' for module in modules.modules(): #print(module)