文件名称: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
文件名称: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