Closed xlnn closed 8 months ago
Hi @xlnn, I appreciate your efforts with our Agent-Swin model. However, attempting to load weights meant for Swin will naturally lead to a mismatch. To ensure compatibility, please use the pretrained weights that we provide.
感谢您出色的工作,我还有一个问题,打扰您一下:我将swin transformer的注意力换成了您的AgentAttention:
我使用了预训练权重:/home/class1/work/modify/G/checkpoints/swin_tiny_patch4_window7_224.pth https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_tiny_patch4_window7_224.pth 数据集使用了coco格式的数据,替换成您的AgentAttention后,发生了如下错误:
您知道如何解决吗?谢谢您!