Skip to content

Commit 754290e

Browse files
author
zhangchen76
committed
Minor.
1 parent e49f403 commit 754290e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

run_pruning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ def main():
166166
neuron_mask.grad = None
167167
model.zero_grad()
168168

169-
# Normalize & merge expressive score and knowledgable score.
169+
# Normalize score.
170170
norm_per_layer = torch.pow(torch.pow(head_score, 2).sum(-1), 0.5)
171171
head_score /= norm_per_layer.unsqueeze(-1) + 1e-7
172172
norm_per_layer = torch.pow(torch.pow(neuron_score, 2).sum(-1), 0.5)

0 commit comments

Comments
 (0)