from local

This commit is contained in:
dohe0342 2023-05-24 13:23:39 +09:00
parent 9c72b903f9
commit ea56d99e73
2 changed files with 2 additions and 1 deletions

View File

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