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
c2386aee5c
commit
bc4c4e8f4e
Binary file not shown.
@ -1588,7 +1588,8 @@ def run_adapter(rank, world_size, args, wb=None):
|
|||||||
adapter_param = []
|
adapter_param = []
|
||||||
for enum, (n, p) in enumerate(model.named_parameters()):
|
for enum, (n, p) in enumerate(model.named_parameters()):
|
||||||
#print(enum, n, p)
|
#print(enum, n, p)
|
||||||
if 'encoder.encoders.layer_norm' in n or 'encoder.encoders.final_proj' in n or 'encoder.output_layer' in n or 'joiner' in n or 'simple' in n or 'ctc' in n:
|
if 'encoder.encoders.layer_norm' in n or 'encoder.encoders.final_proj' in n or 'encoder.output_layer' in n or 'joiner' in n or 'simple' in n or 'ctc' in n or '11' in n:
|
||||||
|
print(n)
|
||||||
adapter_names.append(n)
|
adapter_names.append(n)
|
||||||
adapter_param.append(p)
|
adapter_param.append(p)
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user