Skip to content

开启spatial_freq_decompose后报错 #22

Description

@cocovia

我在测试FDConv的时候,输入张量形状为(16, 128, 64, 64),配置为:FDConv(in_channels=128, out_channels=128, kernel_num=64, kernel_size=3, padding=1, bias=True,spatial_freq_decompose=True),此时报错,该如何解决
File "d:\code\ultralytics-main\ultralytics\nn\addmodules\FDConv.py", line 692, in forward
aggregate_weight = spatial_attention * channel_attention * filter_attention * adaptive_weights * hr_att ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~
RuntimeError: The size of tensor a (256) must match the size of tensor b (128) at non-singleton dimension 3

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions