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
dc00275d63
commit
43954f834b
Binary file not shown.
@ -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):
|
||||
lora_modules.append(LoRAHook(module))
|
||||
lora_modules.append(LoRAHook(modules))
|
||||
#for module in modules.modules():
|
||||
# if rank == 0: logging.info(module)
|
||||
#print(module)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user