摘要: 网络结构定义 import torch import torch.nn as nn import torch.nn.functional as F import torch_pruning as tp from torchvision.datasets import CIFAR10 from tor 阅读全文
posted @ 2022-12-04 21:56 牛犁heart 阅读(1139) 评论(0) 推荐(0) 编辑
摘要: Torch-Pruning 通道剪枝网络实现加速的工作。 Torch pruning是进行结构剪枝的pytorch工具箱,和pytorch官方提供的基于mask的非结构化剪枝不同,工具箱移除整个通道剪枝,自动发现层与层剪枝的依赖关系,可以处理Densenet、ResNet和DeepLab 特性 卷积 阅读全文
posted @ 2022-12-04 18:47 牛犁heart 阅读(2092) 评论(0) 推荐(0) 编辑