This commit is contained in:
游雁 2024-06-13 19:35:57 +08:00
parent 5ebcba3d61
commit 70df8c97dd

View File

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