from local
This commit is contained in:
parent
aa20fc3d97
commit
3ef1ae9180
Binary file not shown.
@ -1144,7 +1144,7 @@ def run(rank, world_size, args, wb=None):
|
|||||||
|
|
||||||
if wb is None:
|
if wb is None:
|
||||||
optimizer_enc = ScaledAdam(
|
optimizer_enc = ScaledAdam(
|
||||||
model.parameters(),
|
enc_param,
|
||||||
lr=params.base_lr,
|
lr=params.base_lr,
|
||||||
clipping_scale=2.0,
|
clipping_scale=2.0,
|
||||||
parameters_names=parameters_names,
|
parameters_names=parameters_names,
|
||||||
|
|||||||
Reference in New Issue
Block a user