PANPEIWEN / ABC

ABC: Attention with Bilinear Correlation for Infrared Small Target Detection
MIT License
29 stars 3 forks source link

代码调试问题 #17

Closed 170zhd closed 2 months ago

170zhd commented 5 months ago

尊敬的作者您好: 我在跑NUAA这个数据集时遇到下列问题,RuntimeError: stack expects each tensor to be equal size, but got [1, 512, 512] at entry 0 and [4, 512, 512] at entry 5。想问您一下,这个问题该如何解决,可以跑通,但是在第一轮结束的时候,出现上述情况,config都是按照您的配置进行的,没有经过任何更改。希望在您有空的时候给我一个答复。不胜感激。

![Uploading 捕获.PNG…]()