from local

This commit is contained in:
dohe0342 2022-12-10 15:59:57 +09:00
parent 38b5ef3ead
commit 85848f50fd
2 changed files with 2 additions and 2 deletions

View File

@ -1239,8 +1239,8 @@ def run(rank, world_size, args, wb=None):
parameters_names=[dec_names],
)
scheduler_enc = Eden(optimizer_enc, params.lr_batches*params.accum_grads, params.lr_epochs)
scheduler_dec = Eden(optimizer_dec, params.lr_batches*params.accum_grads, params.lr_epochs)
scheduler_enc = Eden(optimizer_enc, params.lr_batches, params.lr_epochs)
scheduler_dec = Eden(optimizer_dec, params.lr_batches, params.lr_epochs)
optimizer = [optimizer_enc, optimizer_dec]
scheduler = [scheduler_enc, scheduler_dec]