from local

This commit is contained in:
dohe0342 2022-12-10 16:05:25 +09:00
parent f0a939469e
commit 573a79b08d
2 changed files with 2 additions and 2 deletions

View File

@ -1229,8 +1229,8 @@ def run(rank, world_size, args, wb=None):
dec_names.append(n)
dec_param.append(p)
logging.info(enc_names)
if rank == 0:
print(enc_names)
optimizer_enc = ScaledAdam(
enc_param,
lr=params.peak_enc_lr,