yolov5-6.1,改进中遇到的问题
您好，我是一名初学者。在尝试复现您的文章《改进YOLOv5 | Stand-Alone Self-Attention | 针对视觉任务的独立自注意力层 | 搭建纯注意力FPN+PAN结构》，运行时，在AttentionConv类和AttentionStem类末尾的“init”处报错，想询问一下解决方案。使用的是yolov5-6.1
你只需要在common.py中加入如下代码即可：
from torch.nn import init
您好，我是一名初学者。在尝试复现您的文章《改进YOLOv5 | Stand-Alone Self-Attention | 针对视觉任务的独立自注意力层 | 搭建纯注意力FPN+PAN结构》，运行时，在AttentionConv类和AttentionStem类末尾的“init”处报错，想询问一下解决方案。使用的是yolov5-6.1
你只需要在common.py中加入如下代码即可：
from torch.nn import init