LINC-BIT's picture
Upload 1912 files
b84549f verified
## 20230527
only prune to_v (so should divide to_qkv to to_qkv and to_v, and only prune to_v)
attn (1, 12, 197, 197) * pruned_v (1, 12, 197, int(64\*s)) = out1 (1, 12, 197, int(64\*s))
out1 rearange to out2 (1, 197, int(768\*s))
out2 (1, 197, int(768\*s)) * pruned mlp_fc1 (int(768\*s), int(3072\*s))