from local

This commit is contained in:
dohe0342 2022-12-10 14:32:28 +09:00
parent f6be3cefe5
commit 37412014b1
2 changed files with 3 additions and 3 deletions

View File

@ -1185,16 +1185,16 @@ def run(rank, world_size, args, wb=None):
logging.info(f"enc name len = {len(enc_names)}")
logging.info(f"enc param len = {len(enc_param)}")
logging.info(model.parameters())
#logging.info(model.parameters())
optimizer_enc = ScaledAdam(
enc_param,
*enc_param,
lr=params.peak_enc_lr,
clipping_scale=None,
parameters_names=enc_names,
)
optimizer_dec = ScaledAdam(
dec_param,
*dec_param,
lr=params.peak_dec_lr,
clipping_scale=None,
parameters_names=dec_names,