from local
This commit is contained in:
parent
a66d70da6e
commit
9176d4fe4e
Binary file not shown.
@ -1145,7 +1145,7 @@ def run(rank, world_size, args, wb=None):
|
|||||||
if wb is None:
|
if wb is None:
|
||||||
optimizer_enc = ScaledAdam(
|
optimizer_enc = ScaledAdam(
|
||||||
dec_param,
|
dec_param,
|
||||||
lr=params.base_lr,
|
lr=params.peak_enc_lr,
|
||||||
clipping_scale=5.0,
|
clipping_scale=5.0,
|
||||||
parameters_names=dec_names,
|
parameters_names=dec_names,
|
||||||
)
|
)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user