from local

This commit is contained in:
dohe0342 2022-12-10 14:22:02 +09:00
parent bfe291a40a
commit 43cde32790
2 changed files with 2 additions and 0 deletions

View File

@ -1183,6 +1183,8 @@ def run(rank, world_size, args, wb=None):
dec_names.append(n)
dec_param.append(p)
logging.info(f"{len(enc_names)}")
logging.info(f"{len(enc_param)}")
optimizer_enc = ScaledAdam(
enc_param,
lr=params.peak_enc_lr,