from local

This commit is contained in:
dohe0342 2023-01-03 17:49:56 +09:00
parent 86cc17624c
commit ee475600f2
2 changed files with 1 additions and 1 deletions

View File

@ -1538,7 +1538,7 @@ def run_adapter(rank, world_size, args, wb=None):
adapter_names = []
adapter_param = []
for n, p in model.named_parameters():
if 'adapters' in n or 'joiner' in n or 'simple' in n:
if 'adapters' in n or 'joiner' in n or 'simple' in n or 'ctc' in n:
adapter_names.append(n)
adapter_param.append(p)
#else: