weixin_37911219 2023-12-07 09:33
浏览 15
已结题

tp.pruner.MagnitudePruner报错KeyError，如何解决？

在使用torch-pruning给siamRPN++模型剪枝时，构建剪枝器的过程中报错如下：Parameter containing: tensor([ 0.2020, 0.3365, -0.5385], device='cuda:0', requires_grad=True)

img

一步一步调试进去，会进入model的forward函数，都能顺利通过，没有报错，但是调试出来回到构建剪枝器这里，依然报错。

有没有小伙伴知道这是怎么回事呢？求指导！感谢~

  • 写回答

0条回答 默认 最新

    报告相同问题？

    • 基于torch_pruning库实现两种剪枝器：MagnitudePruner和Slimming（BNScalePruner）
      2025-08-16 17:29
      ghx3110的博客 本文主要基于torch_pruning剪枝工具借鉴实现了MagnitudePruner权值剪枝器和利用BN层scale系数的Slimming剪枝器。
    • 深度学习中的模型剪枝工具Torch-Pruning的使用
      2025-07-19 18:57
      fengbingchun的博客 layers: {ignored_layers}") example_inputs = torch.randn(1, 3, 224, 224) # build network pruners importance = tp.importance.MagnitudeImportance(p=1) pruner = tp.pruner.MagnitudePruner( model, ...
    • tensorflow c++加载checkpoint model.meta model.index报错解决及转cv::Mat为Tensor
      2019-07-04 16:57
      元气少女缘结神的博客 throw runtime_error("Error loading checkpoint from " + checkpointPath + ": " + status.ToString()); } cout !"; //read image for prediction... string image_path= "/media/root/Ubuntu311/projects/...
    • torch-pruning剪枝
      2025-11-11 16:31
      柯瑞奥的博客 类tp.importance.Imprtance要求实现一个接口__call__，其中输入参数是一个group，它包含了多个相互耦合的层；输出是一个一维的重要性得分向量，其含义是每个通道的重要性，所以它的维度和通道数通常是相同的。
    • 模型剪枝 | Torch-Pruning 库入门级使用介绍
      2024-07-15 02:15
      万里鹏程转瞬至的博客 iterative_steps = 5 # progressive pruning pruner = tp.pruner.MagnitudePruner( model, example_inputs, importance=imp, iterative_steps=iterative_steps, ch_sparsity=0.5, # remove 50% channels, ResNet18 ...
    • yolov8剪枝实践
      2023-10-11 18:03
      Ctrl_Cver的博客 v2 for bn_key in ['weight', 'bias', 'running_mean', 'running_var']: old_bn = state_dict[f'cv1.bn.{bn_key}'] state_dict_v2[f'cv0.bn.{bn_key}'] = old_bn[:half_channels] state_dict_v2[f'cv1.bn.{bn_key}']...
    • entity-pruner-3.1.0.jar
      2022-03-14 22:38
      jar包，官方版本，自测可用
    • entity-pruner-3.1.0-sources.jar
      2022-03-14 22:39
      jar包，官方版本，自测可用
    • 【6s965-fall2022】剪枝✂pruningⅠ
      2023-01-19 20:09
      代码缝合怪的博客 return masks prune the pre-trained model 所有的卷积层的稀疏度都设为0.80 # all the conv layers with the same sparsity sparsity = 0.80 pruner = FineGrainedPruner(model, sparsity) pruner.apply(model) ...
    • yolov8 剪枝源码（集成多种剪枝策略）
      2024-03-20 21:12
      支持以下的剪枝方法，代码一键运行，并配有md文档说明: (1) lamp 剪枝 (2) slimming 剪枝 (3) group slimming 剪枝 (4) group hessian 剪枝 (5) Taylor 剪枝 (6）Regularization 剪枝 等等
    • readme.docx
      2025-02-11 15:08
      随着人工智能领域的迅猛发展，大型语言模型（LLM）的训练和微调技术日新月异。为了让这些模型更加高效、精简以及具备更好的泛化能力，研究人员提出了多种策略和方法。本文将详细介绍几种当前流行的方法，包括基于多...
    • hive常用参数优化
      2024-08-20 20:10
      锦衣当国的博客 // 列裁剪 SET hive.optimize.pruner = true; // 分区裁剪 SET hive.groupby.skewindata = true; // groupby数据倾斜优化 SET hive.exec.mode.local.auto = true; // 本地执行 SET hive.exec.mode.local.auto.input...
    • 超参数自动化调优指南：Optuna vs. Ray Tune 对比评测
      2025-09-14 03:09
      九章云极AladdinEdu的博客 随着自动化超参数优化（Hyperparameter Optimization, HPO）工具的发展，我们现在可以更智能、更高效地找到最佳超参数配置。 本文将深入对比两个主流的超参数优化框架：Optuna 和 Ray Tune。通过详细的代码示例、...
    • LLM-pruner源码解析
      2024-11-27 23:51
      An_ich的博客 定制pruner，忽略的层 from .pruner import function 刚开始的编辑器 更新之后的编辑器 按照编辑器中的值获取pruner输入通道的函数，定制里面是空值 按照编辑器中的值获取pruner输出通道的函数，定制里面是空值 每个...
    • 剖析低成本AI金融模型压缩技术，促进AI人工智能金融创新
      2025-06-06 13:30
      AI智能探索者的博客 金融行业对AI模型的需求正面临“冰火两重天”：一方面，大语言模型（如金融领域微调的LLM）、深度神经网络（如用于风控的DNN）在复杂任务（如舆情分析、信用评分）中表现卓越；另一方面，金融业务对实时性（如高频...
    • AI 大模型压缩技术如何让你省下千万成本？？
      2025-07-07 15:45
      陈乔布斯的博客 模型压缩技术助力大模型高效部署边缘设备 面对大模型云端部署的高成本与延迟问题，2025年主流压缩技术...Gartner预测2025年65%企业AI模型将采用压缩技术部署边缘设备，较2023年增长240%。典型电商案例显示，该技术年省
    • NNI调试记录-Pruning
      2023-12-26 16:31
      @daviiid的博客 此外还有其他的，如Taylor FO Weight Pruner，Linear Pruner，AGP Pruner，Movement Pruner，基本都大同小异，感兴趣的可以都去尝试一下； 剪枝流程 定义模型，优化器等 # define the model model = TorchModel()....
    • pruner
      2021-03-21 18:27
      护手修枝剪Gauntlet修剪器是一种旨在减少导致P4编译器崩溃或异常行为的P4程序的大小的工具。该工具旨在与结合使用。该修枝剪目前支持与崩溃（退出代码1）或语义翻译错误（退出代码20，定义程序修剪）。...
    • AI原生NLP应用：如何提升模型推理速度？
      2025-09-12 17:15
      AIGC应用创新大全的博客 在AI原生NLP应用（如实时聊天机器人、语义搜索、智能文档助手）中，推理速度是决定用户体验和商业价值的核心指标。想象一下：当用户发送一条消息后，需要等待5秒才能得到回复，这样的应用注定无法留存用户；而如果能...
    • PyPI 官网下载 | phylopypruner-0.3.1.tar.gz
      2022-01-14 19:06
      "Phylo"通常与生物进化和系统发育分析相关，而"pruner"则暗示了该工具可能用于减少或优化大型进化树，删除不必要或不相关的分支，以提高分析效率或简化结果。 在Python中，这样的库通常是通过`setup.py`文件进行...
    • 没有解决我的问题, 去提问

    问题事件

    • 系统已结题 12月15日
    • 创建了问题 12月7日