XX的杰子 2022-12-24 23:09 采纳率: 85.7%
浏览 30
已结题

yolo用hook取中间特征

hooks.append(model.children._modules["17"].register_forward_hook(get_activation("s_f1")))，我想问一下这句代码是可以取出第17层的输出结果吧，它怎么在单机单卡可以训练，但单机多卡就报错呢。torch.nn.modules.module.ModuleAttributeError: 'DistributedDataParallel' object has no attribute 'model'，这个错该怎么解决呢。谢谢大神指导

  • 写回答

1条回答 默认 最新

  • |__WhoAmI__| 2022-12-25 14:35
    关注

    在使用单机多卡训练的情况下，由于模型的结构被封装在了DistributedDataParallel模块中，因此在访问模型的层时应该使用model.module.children而不是model.children。在上述代码中，应该将行修改为：

    hooks.append(model.module.children._modules["17"].register_forward_hook(get_activation("s_f1")))

    这样就可以在单机多卡训练的情况下访问模型的层并正常使用钩子函数了。
    望采纳

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题？

问题事件

  • 系统已结题 1月13日
  • 已采纳回答 1月5日
  • 创建了问题 12月24日

悬赏问题

  • ¥15 我的代码无法在vc++中运行呀，错误很多
  • ¥50 求一个win系统下运行的可自动抓取arm64架构deb安装包和其依赖包的软件。
  • ¥60 fail to initialize keyboard hotkeys through kernel.0000000000
  • ¥30 ppOCRLabel导出识别结果失败
  • ¥15 Centos7 / PETGEM
  • ¥15 csmar数据进行spss描述性统计分析
  • ¥15 各位请问平行检验趋势图这样要怎么调整？说标准差差异太大了
  • ¥15 delphi webbrowser组件网页下拉菜单自动选择问题
  • ¥15 wpf界面一直接收PLC给过来的信号，导致UI界面操作起来会卡顿
  • ¥15 init i2c:2 freq:100000[MAIXPY]: find ov2640[MAIXPY]: find ov sensor是main文件哪里有问题吗