from local

This commit is contained in:
dohe0342 2022-12-26 13:57:41 +09:00
parent 9cbe36823a
commit e63b77a6ad
2 changed files with 7 additions and 0 deletions

View File

@ -1521,6 +1521,13 @@ def run_adapter(rank, world_size, args, wb=None):
if 'adapters' in n:
adapter_names.append(n)
adapter_param.append(p)
optimizer_adapter = ScaledAdam(
adapter_,
lr=params.peak_enc_lr,
clipping_scale=None,
parameters_names=[enc_names],
)
if params.multi_optim:
logging.info("Using seperate optimizers over encoder, decoder ...")