from local

This commit is contained in:
dohe0342 2023-05-24 13:29:30 +09:00
parent 7e06957879
commit 2c8786089e
2 changed files with 3 additions and 1 deletions

View File

@ -1594,7 +1594,9 @@ def run_adapter(rank, world_size, args, wb=None):
print('-'*30)
for i, module in enumerate(model.modules()):
if isinstance(module, fairseq.modules.multihead_attention.MultiheadAttention):
print(module)
for m in module.modules():
print(m)
print('-'*30)
exit()