This commit is contained in:
游雁 2024-06-13 16:22:09 +08:00
parent 6ca0b838d4
commit a1e18f808e

View File

@ -422,7 +422,7 @@ class LLMASR2(nn.Module):
layer_id = int(name[beg + 1 : end - 1])
if layer_id < freeze_layer_num:
param.requires_grad = False
else:
elif not name.startswith("audio_encoder.ln_post"):
param.requires_grad = False
else:
param.requires_grad = False