pytorch-topk.py

时间:2024-01-15 05:38:46
【文件属性】:

文件名称:pytorch-topk.py

文件大小:7KB

文件格式:PY

更新时间:2024-01-15 05:38:46

pytorch 剪枝 nlp fpga

以MNIST数据集为例,对LSTM的权重矩阵实现top-k剪枝(7,2),介绍了如何在pytorch框架下实现top-k剪枝。详细讲解参考这篇博客:https://blog.csdn.net/kuan__/article/details/111665163


网友评论