from local

This commit is contained in:
dohe0342 2022-12-30 12:59:14 +09:00
parent 1817b0a18e
commit 3cf918bbac
3 changed files with 3 additions and 0 deletions

View File

@ -1121,6 +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)
if params.multi_optim:
cur_enc_lr = scheduler_enc.get_last_lr()[0]
cur_dec_lr = scheduler_dec.get_last_lr()[0]