from local
This commit is contained in:
parent
7e06957879
commit
2c8786089e
Binary file not shown.
@ -1594,7 +1594,9 @@ def run_adapter(rank, world_size, args, wb=None):
|
|||||||
print('-'*30)
|
print('-'*30)
|
||||||
for i, module in enumerate(model.modules()):
|
for i, module in enumerate(model.modules()):
|
||||||
if isinstance(module, fairseq.modules.multihead_attention.MultiheadAttention):
|
if isinstance(module, fairseq.modules.multihead_attention.MultiheadAttention):
|
||||||
print(module)
|
for m in module.modules():
|
||||||
|
print(m)
|
||||||
|
|
||||||
print('-'*30)
|
print('-'*30)
|
||||||
|
|
||||||
exit()
|
exit()
|
||||||
|
|||||||
Reference in New Issue
Block a user