我在测试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
我在测试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