from local

This commit is contained in:
dohe0342 2022-12-10 13:46:52 +09:00
parent 7d67bd90b5
commit a66d70da6e
2 changed files with 2 additions and 2 deletions

View File

@ -1144,10 +1144,10 @@ def run(rank, world_size, args, wb=None):
if wb is None: if wb is None:
optimizer_enc = ScaledAdam( optimizer_enc = ScaledAdam(
enc_param, dec_param,
lr=params.base_lr, lr=params.base_lr,
clipping_scale=5.0, clipping_scale=5.0,
parameters_names=parameters_names, parameters_names=dec_names,
) )
optimizer_dec = Eve(dec_param, lr=params.peak_dec_lr) optimizer_dec = Eve(dec_param, lr=params.peak_dec_lr)