from local

This commit is contained in:
dohe0342 2022-12-30 13:04:26 +09:00
parent 3cf918bbac
commit 8b6b4749f4
2 changed files with 3 additions and 3 deletions

View File

@ -1121,9 +1121,9 @@ def train_one_epoch(
)
if batch_idx % (params.log_interval*params.accum_grads) == 0:
for n, p in model.named_parameters():
if 'adapter' in n:
print(p)
#for n, p in model.named_parameters():
# if 'adapter' in n:
# print(p)
if params.multi_optim:
cur_enc_lr = scheduler_enc.get_last_lr()[0]
cur_dec_lr = scheduler_dec.get_last_lr()[0]