关于使用`fused_window_process`提高速度的说明及其在小规模任务中所需的必要性
作者: Qingyu-Fan创建于 2024年10月16日更新于 2025年2月11日
你好,感谢您的出色工作! 我有一个问题要问一下 `fused_window_process`。在将窗口过程集成到 CUDA 文件中后,速度提升是否显著?您能否提供一些量化数据来说明性能提升? 此外,对于规模较小的任务,是否需要使用窗口过程,还是使用 `torch.roll` 的默认实现会更好? 期待您的回复!
内容来源: microsoft/Swin-Transformer