from local

This commit is contained in:
dohe0342 2022-12-10 14:36:54 +09:00
parent 57b79bf164
commit b8061009ab
2 changed files with 2 additions and 2 deletions

View File

@ -1184,13 +1184,13 @@ def run(rank, world_size, args, wb=None):
dec_param.append(p)
optimizer_enc = ScaledAdam(
params=*enc_param,
*enc_param,
lr=params.peak_enc_lr,
clipping_scale=None,
parameters_names=enc_names,
)
optimizer_dec = ScaledAdam(
params=*dec_param,
*dec_param,
lr=params.peak_dec_lr,
clipping_scale=None,
parameters_names=dec_names,