from local
This commit is contained in:
parent
ada2b0848d
commit
bfe291a40a
Binary file not shown.
@ -1185,13 +1185,13 @@ def run(rank, world_size, args, wb=None):
|
|||||||
|
|
||||||
optimizer_enc = ScaledAdam(
|
optimizer_enc = ScaledAdam(
|
||||||
enc_param,
|
enc_param,
|
||||||
lr=param.peak_enc_lr,
|
lr=params.peak_enc_lr,
|
||||||
clipping_scale=None,
|
clipping_scale=None,
|
||||||
parameters_names=enc_names,
|
parameters_names=enc_names,
|
||||||
)
|
)
|
||||||
optimizer_dec = ScaledAdam(
|
optimizer_dec = ScaledAdam(
|
||||||
dec_param,
|
dec_param,
|
||||||
lr=param.peak_dec_lr,
|
lr=params.peak_dec_lr,
|
||||||
clipping_scale=None,
|
clipping_scale=None,
|
||||||
parameters_names=dec_names,
|
parameters_names=dec_names,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user