mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-03 06:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
marcoyang 2024-02-22 20:54:20 +08:00
parent e3a8f14969
commit 86a3a2f73b
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
egs/librispeech/ASR/zipformer_adapter/train.py
View File

@ -1018,6 +1018,12 @@ def train_one_epoch(
be set to 0.
"""
model.train()
# set modules except adapters to eval mode
for name, m in model.named_modules():
if "adapter" in name:
m.training = True
else:
m.training = False
tot_loss = MetricsTracker()
@ -1159,12 +1165,6 @@ def train_one_epoch(
valid_dl=valid_dl,
world_size=world_size,
)
model.train()
for name, m in model.named_modules():
if "adapter" in name:
m.training = True
else:
m.training = False
logging.info(
f"Validation on {valid_set}: Epoch {params.cur_epoch}, validation: {valid_info}"
)
@ -1175,6 +1175,8 @@ def train_one_epoch(
valid_info.write_summary(
tb_writer, f"train/{valid_set}_valid_", params.batch_idx_train
)
model.train()
# set modules except adapters to eval mode
for name, m in model.named_modules():
if "adapter" in name:
m.training = True