问题描述

  • 在使用mmcv的attention时，遇到该问题
    在这里插入图片描述

问题原因

  • 在debug过程中，我发现sampling_locations和attention_weights的num_query维,都是

在这里插入图片描述

解决方法

  • 去往前debug，查看是哪一步操作，使得这里的num_query数为0，我的问题在于mask全为false，导致所有的query都被去除。
# pytorch # 深度学习 # 人工智能
Logo
FlagOS智算系统软件栈

欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

不要死磕CUDA，国内首个Triton技术大会官宣，AI芯片编程迎来新范式

你将听到 Triton 诞生背后的第一性原理与未来演进的宏伟蓝图，获取关于 AI 编译器的最新学术研究报告，了解众智 FlagOS 社区生态的最新进展，更有对增强版 Triton 编译器特性的深度讲解。作为国内专注于 Triton 技术及其生态的开发者盛会，我们诚挚地邀请全球的 AI 研究员、高性能计算工程师、编译器专家、编程语言爱好者，以及所有希望突破 AI 芯片编程复杂性的开发者，与我们一起，

avatar FlagOS智算系统软件栈

具身智能“大脑”谁更强？FlagEval-EmbodiedVerse 11月评测榜单发布

具身智能是连接虚拟与现实、通往通用人工智能（AGI）的关键路径。近日，北京智源人工智能研究院。

avatar FlagOS智算系统软件栈
cover

推出“首个”芯片解耦集合通信技术，众智 FlagOS 支持同/异构全场景互联互通

avatar FlagOS智算系统软件栈