mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
4048f7be65
commit
fecb601ef1
Binary file not shown.
@ -1605,7 +1605,7 @@ def run_adapter(rank, world_size, args, wb=None):
|
|||||||
lora_modules = []
|
lora_modules = []
|
||||||
for modules in model.modules():
|
for modules in model.modules():
|
||||||
if isinstance(modules, fairseq.modules.multihead_attention.MultiheadAttention):
|
if isinstance(modules, fairseq.modules.multihead_attention.MultiheadAttention):
|
||||||
logging.info(modules)
|
if rank == 0: logging.info(modules)
|
||||||
'''
|
'''
|
||||||
for module in modules.modules():
|
for module in modules.modules():
|
||||||
#print(module)
|
#print(module)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user