This commit is contained in:
游雁 2024-06-13 14:26:38 +08:00
parent 5079d54595
commit e42c693f0b

View File

@ -420,6 +420,8 @@ class LLMASR2(nn.Module):
layer_id = int(name[beg + 1 : end - 1])
if layer_id in freeze_layer_num:
param.requires_grad = False
else:
param.requires_grad = False
else:
param.requires_grad = False