from local
This commit is contained in:
parent
a3e4c84069
commit
d9d433fa99
Binary file not shown.
Binary file not shown.
@ -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_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]
|
optimizer = [optimizer_enc, optimizer_dec]
|
||||||
scheduler = [scheduler_enc, scheduler_dec]
|
scheduler = [scheduler_enc, scheduler_dec]
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user