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
22323fa6c8
commit
1b89ed9b45
Binary file not shown.
@ -1606,7 +1606,7 @@ def run_adapter(rank, world_size, args, wb=None):
|
||||
for modules in model.modules():
|
||||
if isinstance(modules, fairseq.modules.multihead_attention.MultiheadAttention):
|
||||
for module in modules.modules():
|
||||
if rank == 0: logging.info(modules)
|
||||
if rank == 0: logging.info(module)
|
||||
#print(module)
|
||||
#if isinstance(module, torch.nn.Linear):
|
||||
# lora_modules.append(LoRAHook(module))
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user