from local

This commit is contained in:
dohe0342 2022-12-10 13:46:21 +09:00
parent aa20fc3d97
commit 3ef1ae9180
2 changed files with 1 additions and 1 deletions

View File

@ -1144,7 +1144,7 @@ def run(rank, world_size, args, wb=None):
if wb is None:
optimizer_enc = ScaledAdam(
model.parameters(),
enc_param,
lr=params.base_lr,
clipping_scale=2.0,
parameters_names=parameters_names,