from local

This commit is contained in:
dohe0342 2022-12-10 14:22:23 +09:00
parent a8e608469b
commit 226d180832
2 changed files with 1 additions and 0 deletions

View File

@ -1185,6 +1185,7 @@ 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)}")
optimizer_enc = ScaledAdam(
enc_param,
lr=params.peak_enc_lr,