python中权重剪枝，低秩分解，量化技术 代码
目录
python中权重剪枝，低秩分解，量化技术 代码
权重剪枝
权重剪枝可以通过PyTorch的torch.nn.utils.prune模块实现。以下是一个简单的例子：
import torch
import torch.nn as nn
import torch.nn.utils.prune as prune
class SimpleNet(nn.Module):
def __init__(self):
super(SimpleNet, self).__init__()
self.conv1 = nn.Conv2d(1, 10, kernel_size=5)
self.fc1 = nn.Linear(10*12*12, 50)
self.fc2 = nn.Linear(50, 10)
def forwar原文地址：https://blog.csdn.net/qq_38998213/article/details/144226786
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！