from local

This commit is contained in:
dohe0342 2023-05-15 13:04:33 +09:00
parent ba4a993c3a
commit 8b80b1ab14
2 changed files with 1 additions and 1 deletions

View File

@ -1588,7 +1588,7 @@ def run_adapter(rank, world_size, args, wb=None):
adapter_param = []
for enum, (n, p) in enumerate(model.named_parameters()):
#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 or '11' 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 or '10' in n:
print(n)
adapter_names.append(n)
adapter_param.append(p)