from local

This commit is contained in:
dohe0342 2022-12-10 14:18:40 +09:00
parent ada2b0848d
commit bfe291a40a
2 changed files with 2 additions and 2 deletions

View File

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