from local

This commit is contained in:
dohe0342 2022-12-10 14:50:01 +09:00
parent a3e4c84069
commit d9d433fa99
3 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

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