当前位置: 首页 > news >正文

在 parse_model 函数中添加了自定义模块支持


第一段代码(已修改版本)

在 parse_model 函数中添加了自定义模块支持:

n = n_ = max(round(n * depth), 1) if n > 1 else n # depth gain

# Custom modules support - Added for RFAConv, HSFPN, HATHead integration
custom_base_module_names = {'RFAConv', 'RFCBAMConv', 'RFABlock'}
is_base_module = m in base_modules or (hasattr(m, '__name__') and m.__name__ in custom_base_module_names)

if is_base_module: # 👈 使用 is_base_module
c1, c2 = ch[f], args[0]
# ... 后续处理

第二段代码(原始版本)

直接检查模块是否在 base_modules 中:

n = n_ = max(round(n * depth), 1) if n > 1 else n # depth gain

if m in base_modules: # 👈 直接使用 m in base_modules
c1, c2 = ch[f], args[0]
# ... 后续处理

修改的意义

第一段代码的修改允许:
1. 扩展自定义模块支持:除了内置的 base_modules,还支持 RFAConv、RFCBAMConv、RFABlock 这些自定义注意力模块
2. 动态检查模块名称:通过 hasattr(m, '__name__') 检查模块是否有名称属性
3. 灵活性更强:可以轻松添加更多自定义模块到 custom_base_module_names 集合中

http://www.zskr.cn/news/52526.html

相关文章:

  • 20232311 2025-2026-1 《网络与系统攻防技术》实验六实验报告
  • android compose viewModel 传参数
  • 奶牛快传服务调整公告
  • 从零实现 REINFORCE/GRPO —— 大模型推理强化微调实践
  • 手撸大模型的分布式训练:深刻理解大模型训练的“起飞”原理
  • 从0到1:揭秘LLM预训练前的海量数据清洗全流程
  • instr在mysql索引中作用是什么
  • Python调用C++代码
  • MySQL EXPLAIN中的key_len:精准掌握索引使用情况
  • AWS云服务深度集成
  • httpd linux 启动
  • Node.js服务稳定性保障:从热更新到高可用体系
  • PG系列:在 ​​psql​​ 客户端中定义参数与动态赋值
  • 欢迎关注我的公众号和B站
  • 11/17
  • linux 下中文字体安装.ttf 格式
  • 2025 年锚具厂家 TOP 企业品牌推荐排行榜,桥梁伸缩缝 / 道路伸缩缝 / 梳齿板伸缩缝推荐这十家公司!
  • 2025-11-17
  • 论文速读 | 2025年11月
  • halt linux
  • hadoop linux 安装
  • 解决罗技M590右键必须用力才能使用的问题
  • sequence 题解
  • 20232410 2025-2026-1 《网络与系统攻防技术》实验六实验报告
  • FastAPI Test Project
  • React Scheduler(调度器)
  • 2025年11月学习机榜单:双线提分机型领衔,十大高性价比之选
  • vue2和vue3声明式和命令时的区别
  • 3D 文件类型,怎么在线查看编辑STL/AMF/OBJ/stp/fbx/ply转换
  • 022304105叶骋恺数据采集第三次作业